Update utils.py

utils.py

@@ -48,13 +48,13 @@ class make_im():
             os.mkdir(f'{self.root}/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)
-        img=PIL.Image.frombytes(self.pimg)
+        #img=PIL.Image.frombytes(self.pimg)
         #matplotlib.image.imsave(f"{self.root}/images/{name}_mask.png", self.mask, cmap='gray')
         #matplotlib.image.imsave(f"{self.root}/images/{name}_epwt.png", self.epwtimg)
         #matplotlib.image.imsave(f"{self.root}/images/{name}_prob.png", self.pimg)
         buffer = io.BytesIO()
-        img.save(buffer, format='PNG')
-        #buffer.write(self.pimg)
+        #img.save(buffer, format='PNG')
+        buffer.write(self.pimg)
         #buffer.seek(0)
         return buffer
     #img = get_random_image(img_size)  #returns numpy array