Update utils.py

utils.py

@@ -27,12 +27,12 @@ class make_im():
             cmap = ea
             self.epwtimg = ColoredPath(self.epwtpath, self.mask.shape).get_colored_path(cmap)
             self.pimg = ColoredPath(self.ppath, self.mask.shape).get_colored_path(cmap)
-            if not os.path.isdir(f"./images/{ea}"):
-                os.mkdir(f'./images/{ea}')
+            if not os.path.isdir(f"./image_alls/{ea}"):
+                os.mkdir(f'./images_all/{ea}')
             else: pass
-            matplotlib.image.imsave(f"./images/{ea}/{ea}_mask.png", self.mask, cmap='gray')
-            matplotlib.image.imsave(f"./images/{ea}/{ea}_epwt.png", self.epwtimg)
-            matplotlib.image.imsave(f"./images/{ea}/{ea}_prob.png", self.pimg)
+            matplotlib.image.imsave(f"./images_all/{ea}/{ea}_mask.png", self.mask, cmap='gray')
+            matplotlib.image.imsave(f"./images_all/{ea}/{ea}_epwt.png", self.epwtimg)
+            matplotlib.image.imsave(f"./images_all/{ea}/{ea}_prob.png", self.pimg)
     
     def run_1(self,img_size=[128,128],cmap='flag_r',key=0):
         randvar = random.randint(2,10)
@@ -40,14 +40,14 @@ class make_im():
         self.epwtpath = EPWTPath(self.mask).get_path()
         sel.ppath = ProbabilisticPath(self.mask).get_path()
     
-        if not os.path.isdir(f"./user"):
-            os.mkdir(f'./user')
+        if not os.path.isdir(f"./images"):
+            os.mkdir(f'./images')
         self.epwtimg = ColoredPath(self.epwtpath, self.mask.shape).get_colored_path(cmap)
         self.pimg = ColoredPath(self.ppath, self.mask.shape).get_colored_path(cmap)
     
-        matplotlib.image.imsave(f"./user/{key}_mask.png", self.mask, cmap='gray')
-        matplotlib.image.imsave(f"./user/{key}_epwt.png", self.epwtimg)
-        matplotlib.image.imsave(f"./user/{key}_prob.png", self.pimg)
+        matplotlib.image.imsave(f"./images/{key}_mask.png", self.mask, cmap='gray')
+        matplotlib.image.imsave(f"./images/{key}_epwt.png", self.epwtimg)
+        matplotlib.image.imsave(f"./images/{key}_prob.png", self.pimg)
         return self.pimg
     #img = get_random_image(img_size)  #returns numpy array
     #print(img)