path changes

Author: ishaan95

src/__pycache__/maps.cpython-39.pyc

@@ -76,7 +76,7 @@ def generate_patches(self, mapReader, image_groups=3):
                 for x in range(self.window_size[0])])
 
                 if len(self.samples)==self.sample_group_size:
-                    with open("data/output/SF_group"+str(img_group_number)+".dill", 'wb+') as f:
+                    with open("../data/output/city_patches/dill/sf/"+str(img_group_number)+".dill", 'wb+') as f:
                         dill.dump(self.samples, f)
                         f.close()    
                     print("Image group {} saved in data/output/ !".format(img_group_number))

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')
+
+
 
 

src/read.py

@@ -2,5 +2,5 @@
 import maps
 
 sfMap = maps.MapReader('../data/input/sf_layered.txt')
-mapsDataset = maps.MapsDataset((32, 32), 2, 1280, maps.single_layer_converter) #Third parameter is the group size
+mapsDataset = maps.MapsDataset((64, 64), 2, 1280, maps.single_layer_converter) #Third parameter is the group size
 mapsDataset.generate_patches(sfMap) #This will generate dill files which contain the saved sample lists.