Update utils.py

utils.py

@@ -51,7 +51,7 @@ class make_im():
         #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)
+        matplotlib.image.imsave(f"{self.root}/images/{name}_prob.png", self.pimg)
         buffer = io.BytesIO()
         #img.save(buffer, format='PNG')
         buffer.write(self.pimg)