Spaces:
Running
on
Zero
Running
on
Zero
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)
|