Merge branch 'main' into muktadir

Author: adhocmaster

src/__pycache__/maps.cpython-39.pyc

@@ -99,6 +99,7 @@ def generate_patches(self, mapReader, image_groups=3, outDirectory=None):
                     logging.info(f"Image group {img_group_number} saved in {path}")
                     self.samples.clear()
                     img_group_number+=1
+        pass
 
     def extractSample(self, mapReader, topLeft):
         i = topLeft[0]

src/maps.py

@@ -13,12 +13,12 @@ def __init__(self, directory_path, location='SF'):
 
     def load_group(self, group=0):
         if self.dir_path[-1]=='/':
-            with open(self.dir_path+self.location+"_group"+str(group)+".dill", 'rb') as d:
+            with open(self.dir_path+str(group)+".dill", 'rb') as d:
                 self.data = dill.load(d)
                 d.close()
         else:
             self.dir_path = self.dir_path + "/"
-            with open(self.dir_path+self.location+"_group"+str(group)+".dill", 'rb') as d:
+            with open(self.dir_path+str(group)+".dill", 'rb') as d:
                 self.data = dill.load(d)
                 d.close()
 
@@ -30,10 +30,14 @@ def read_numpy(self):
             self.data = self.data.astype(np.uint8)
         return self.data
 
-reader = ImgGroupReader("data/output/")
-reader.load_group()
-data = reader.read_numpy()
-im = Image.fromarray(data[0])
-im.show()
+sf_groups = 117
+for i in range(sf_groups):
+    reader = ImgGroupReader("../data/output/city_patches/dill/washington")
+    reader.load_group(group=i)
+    data = reader.read_numpy()
+    im = Image.fromarray(data[0])
+    im.save('../data/output/city_patches/png/washington/'+str(i)+'.png')
+
+