Latest experimentations with tiff images

compress_start.py

@@ -42,30 +42,48 @@ def image_extractor(scenes):
 
 if __name__ == '__main__':
 
+    """For boundary cases: Start by grabbing the shape of the images and saving those
+    as variables. Then, if statements for if row == 0 or row == maximum and if
+    col == 0 or col == maximum. Then grab corresponding open pixels. Then proceed to do
+    an and statement that handles the corners"""
+    tiff0 = []
+    tiff1 = []
+    tiff2 = []
+    tiff3 = []
+    
+    files = os.listdir("images")
+    scenes = []
+    for file in files:
+        scene = os.path.join("images", file)
+        sc = os.listdir(scene)
+        for s in sc:
+            if s[-6] == '0':
+                tiff0.append(os.path.join(scene, s))
+            elif s[-6] == '1':
+                tiff1.append(os.path.join(scene, s))
+            elif s[-6] == '2':
+                tiff2.append(os.path.join(scene, s))
+            elif s[-6] == '3':
+                tiff3.append(os.path.join(scene, s))
  
-    image = Image.open("practice_tiff.tiff")
-    image2 = Image.open("practice2.tiff")
-    imarray2 = np.array(image2)
-    imarray = np.array(image)
-    work = imarray2[:,:,0]
-    ind1, ind2 = np.random.randint(0,434), np.random.randint(0,650)
+    diff = []
+    for ii in range(len(tiff1)):
+        image = Image.open(tiff1[ii])
+        image = np.array(image)[1:,:]
+        ar1, ar2 = image.shape
+       
+        ind1, ind2 = np.random.randint(0,ar1), np.random.randint(0,ar2)
         surrounding = []
         for i,j in product(np.arange(-1,2), repeat=2):
             if i == 0 and j == 0:
                 continue
             else:
-            surrounding.append(work[ind1+i, ind1+j])
-    diff = []
+                surrounding.append(image[ind1+i, ind1+j])
         diff.append(np.max(surrounding)-np.min(surrounding))
-    print(surrounding)
     print(diff)
     
-    """For boundary cases: Start by grabbing the shape of the images and saving those
-    as variables. Then, if statements for if row == 0 or row == maximum and if
-    col == 0 or col == maximum. Then grab corresponding open pixels. Then proceed to do
-    an and statement that handles the corners"""
     
-    """scene_names = file_extractor()
+    """scene_names = file_extractor("images")
     images = image_extractor(scene_names)
     im = Image.open(images[0])
     imarray = np.array(im)