Spaces:

0x90e
/

ESRGAN-MANGA

Runtime error

0x90e commited on Dec 5, 2022

Commit

cb317d1

1 Parent(s): af9808e

Test more compression

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -53,4 +53,4 @@ with torch.no_grad():
     output = np.transpose(output[[2, 1, 0], :, :], (1, 2, 0))
     output = (output * 255.0).round()
 print('Finished upscaling, saving image.')
-cv2.imwrite(output_dir, output, [int(cv2.IMWRITE_PNG_COMPRESSION), 5])

     output = np.transpose(output[[2, 1, 0], :, :], (1, 2, 0))
     output = (output * 255.0).round()
 print('Finished upscaling, saving image.')
+cv2.imwrite(output_dir, output, [int(cv2.IMWRITE_PNG_COMPRESSION), 9])