Spaces:
Running
on
T4
Running
on
T4
Zongsheng
commited on
Commit
•
8e0f09c
1
Parent(s):
4b550b9
debug image channle
Browse files
app.py
CHANGED
|
@@ -73,6 +73,7 @@ def predict(im_path, background_enhance, face_upsample, upscale, started_timeste
|
|
| 73 |
restored_image_dir.mkdir()
|
| 74 |
# save the whole image
|
| 75 |
save_path = restored_image_dir / Path(im_path).name
|
|
|
|
| 76 |
util_image.imwrite(image_restored, save_path, chn='rgb', dtype_in='uint8')
|
| 77 |
|
| 78 |
return image_restored, str(save_path)
|
|
|
|
| 73 |
restored_image_dir.mkdir()
|
| 74 |
# save the whole image
|
| 75 |
save_path = restored_image_dir / Path(im_path).name
|
| 76 |
+
image_restored = util_image.bgr2rgb(image_restored)
|
| 77 |
util_image.imwrite(image_restored, save_path, chn='rgb', dtype_in='uint8')
|
| 78 |
|
| 79 |
return image_restored, str(save_path)
|