save as png

segment_utils.py

@@ -25,10 +25,10 @@ def restore_result(croper, category, generated_image):
     restored_image.paste(cropped_generated_image, (croper.origin_start_x, croper.origin_start_y), cropped_square_mask_image)
 
     extension = 'png'
-    if restored_image.mode == 'RGBA':
-        extension = 'png'
-    else:
-        extension = 'jpg'
+    # if restored_image.mode == 'RGBA':
+    #     extension = 'png'
+    # else:
+    #     extension = 'jpg'
     
     if not os.path.exists("output"):
         os.makedirs("output")