Spaces:
Running
Running
Update inference_img.py
Browse files- inference_img.py +1 -49
inference_img.py
CHANGED
|
@@ -1,49 +1,3 @@
|
|
| 1 |
-
Hugging Face's logo
|
| 2 |
-
Hugging Face
|
| 3 |
-
Models
|
| 4 |
-
Datasets
|
| 5 |
-
Spaces
|
| 6 |
-
Community
|
| 7 |
-
Docs
|
| 8 |
-
Enterprise
|
| 9 |
-
Pricing
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
Spaces:
|
| 14 |
-
|
| 15 |
-
AItool
|
| 16 |
-
/
|
| 17 |
-
RIFE_Interpolation
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
like
|
| 21 |
-
0
|
| 22 |
-
|
| 23 |
-
Logs
|
| 24 |
-
App
|
| 25 |
-
Files
|
| 26 |
-
Community
|
| 27 |
-
Settings
|
| 28 |
-
RIFE_Interpolation
|
| 29 |
-
/
|
| 30 |
-
inference_img.py
|
| 31 |
-
|
| 32 |
-
AItool's picture
|
| 33 |
-
AItool
|
| 34 |
-
Create inference_img.py
|
| 35 |
-
b0b5f60
|
| 36 |
-
verified
|
| 37 |
-
3 days ago
|
| 38 |
-
raw
|
| 39 |
-
|
| 40 |
-
Copy download link
|
| 41 |
-
history
|
| 42 |
-
blame
|
| 43 |
-
edit
|
| 44 |
-
delete
|
| 45 |
-
|
| 46 |
-
4.48 kB
|
| 47 |
import os
|
| 48 |
import cv2
|
| 49 |
import torch
|
|
@@ -159,6 +113,4 @@ for i in range(len(img_list)):
|
|
| 159 |
else:
|
| 160 |
success = cv2.imwrite(filename_png, (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
|
| 161 |
print(f"Saving to {filename_png} → success: {success}")
|
| 162 |
-
print("Saving to:", os.path.abspath(filename_png))
|
| 163 |
-
|
| 164 |
-
print("Checkpoint reached!")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import os
|
| 2 |
import cv2
|
| 3 |
import torch
|
|
|
|
| 113 |
else:
|
| 114 |
success = cv2.imwrite(filename_png, (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
|
| 115 |
print(f"Saving to {filename_png} → success: {success}")
|
| 116 |
+
print("Saving to:", os.path.abspath(filename_png))
|
|
|
|
|
|