Update utils.py

utils.py

@@ -48,7 +48,7 @@ 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.open(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)