Spaces:
Running
Running
set limit back
Browse files
app.py
CHANGED
|
@@ -58,6 +58,8 @@ def run_rerun(path_to_video):
|
|
| 58 |
if frame_idx % 3 != 0:
|
| 59 |
continue
|
| 60 |
|
|
|
|
|
|
|
| 61 |
# resize to avoid excessive time spent processing
|
| 62 |
frame = cv2.resize(frame, (640, 480))
|
| 63 |
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
|
@@ -93,7 +95,7 @@ def run_rerun(path_to_video):
|
|
| 93 |
os.remove(path_to_video)
|
| 94 |
|
| 95 |
|
| 96 |
-
@spaces.GPU(duration=
|
| 97 |
def estimate_depth(frame):
|
| 98 |
image = transform(frame)
|
| 99 |
image = image.to(device)
|
|
|
|
| 58 |
if frame_idx % 3 != 0:
|
| 59 |
continue
|
| 60 |
|
| 61 |
+
print("processing frame", frame_idx)
|
| 62 |
+
|
| 63 |
# resize to avoid excessive time spent processing
|
| 64 |
frame = cv2.resize(frame, (640, 480))
|
| 65 |
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
|
|
|
| 95 |
os.remove(path_to_video)
|
| 96 |
|
| 97 |
|
| 98 |
+
@spaces.GPU(duration=20)
|
| 99 |
def estimate_depth(frame):
|
| 100 |
image = transform(frame)
|
| 101 |
image = image.to(device)
|