Update utils.py

utils.py

@@ -27,9 +27,14 @@ class make_im():
             if not os.path.isdir(f"{self.root}/image_alls/{ea}"):
                 os.mkdir(f'{self.root}/images_all/{ea}')
             else: pass
-            matplotlib.image.imsave(f"{self.root}/images_all/{ea}/{ea}_mask.png", self.mask, cmap='gray')
-            matplotlib.image.imsave(f"{self.root}/images_all/{ea}/{ea}_epwt.png", self.epwtimg)
-            matplotlib.image.imsave(f"{self.root}/images_all/{ea}/{ea}_prob.png", self.pimg)
+            buffer = io.BytesIO()
+            buffer.write(self.pimg)
+            buffer.seek(0)
+            #matplotlib.image.imsave(f"{self.root}/images_all/{ea}/{ea}_mask.png", self.mask, cmap='gray')
+            #matplotlib.image.imsave(f"{self.root}/images_all/{ea}/{ea}_epwt.png", self.epwtimg)
+            #matplotlib.image.imsave(f"{self.root}/images_all/{ea}/{ea}_prob.png", self.pimg)
+        return buffer
+
     
     def run_1(self,name='temp',img_size=[128,128],cmap='flag_r',key=0):
         randvar = random.randint(2,10)