Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
42437d6
1
Parent(s):
81d4d32
app.py
CHANGED
|
@@ -77,31 +77,33 @@ PROMPT = 'depth'
|
|
| 77 |
NEGATIVE_PROMPT = ''
|
| 78 |
|
| 79 |
example_inputs = [
|
|
|
|
| 80 |
["examples/1.mp4", "1.3B", 5, 3],
|
| 81 |
-
["examples/
|
| 82 |
-
|
| 83 |
-
|
|
|
|
| 84 |
["examples/7.mp4", "1.3B", 5, 3],
|
| 85 |
["examples/8.mp4", "1.3B", 5, 3],
|
| 86 |
["examples/9.mp4", "1.3B", 5, 3],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
["examples/10.mp4", "1.3B", 5, 3],
|
| 88 |
["examples/30.mp4", "1.3B", 5, 3],
|
|
|
|
| 89 |
["examples/31.mp4", "1.3B", 5, 3],
|
| 90 |
["examples/32.mp4", "1.3B", 5, 3],
|
| 91 |
-
|
| 92 |
["examples/35.mp4", "1.3B", 5, 3],
|
| 93 |
["examples/36.mp4", "1.3B", 5, 3],
|
| 94 |
["examples/39.mp4", "1.3B", 5, 3],
|
| 95 |
["examples/40.mp4", "1.3B", 5, 3],
|
| 96 |
-
|
| 97 |
-
["examples/178db6e89ab682bfc612a3290fec58dd.mp4", "1.3B", 5, 3],
|
| 98 |
-
["examples/1b0daeb776471c7389b36cee53049417.mp4", "1.3B", 5, 3],
|
| 99 |
-
["examples/8a6dfb8cfe80634f4f77ae9aa830d075.mp4", "1.3B", 5, 3],
|
| 100 |
-
["examples/69230f105ad8740e08d743a8ee11c651.mp4", "1.3B", 5, 3],
|
| 101 |
-
["examples/b1f1fa44f414d7731cd7d77751093c44.mp4", "1.3B", 5, 3],
|
| 102 |
-
["examples/b68045aa2128ab63d9c7518f8d62eafe.mp4", "1.3B", 5, 3],
|
| 103 |
-
|
| 104 |
-
["examples/DJI_20250912163642_0003_D.mp4", "1.3B", 5, 3],
|
| 105 |
["examples/DJI_20250912164311_0007_D.mp4", "1.3B", 5, 3],
|
| 106 |
]
|
| 107 |
|
|
@@ -271,10 +273,7 @@ def get_model(model_size):
|
|
| 271 |
raise ValueError(f"Unsupported model size: {model_size}")
|
| 272 |
|
| 273 |
|
| 274 |
-
|
| 275 |
-
flipped = points.copy()
|
| 276 |
-
flipped[:, 2] = -flipped[:, 2] # 上下翻转点云
|
| 277 |
-
return flipped
|
| 278 |
|
| 279 |
|
| 280 |
|
|
@@ -427,7 +426,7 @@ def process_video(
|
|
| 427 |
|
| 428 |
|
| 429 |
# ==== 新增:上下翻转点云 ====
|
| 430 |
-
points =
|
| 431 |
# =========================
|
| 432 |
|
| 433 |
|
|
|
|
| 77 |
NEGATIVE_PROMPT = ''
|
| 78 |
|
| 79 |
example_inputs = [
|
| 80 |
+
|
| 81 |
["examples/1.mp4", "1.3B", 5, 3],
|
| 82 |
+
["examples/33.mp4", "1.3B", 5, 3],
|
| 83 |
+
|
| 84 |
+
|
| 85 |
+
# ["examples/5.mp4", "1.3B", 5, 3],
|
| 86 |
["examples/7.mp4", "1.3B", 5, 3],
|
| 87 |
["examples/8.mp4", "1.3B", 5, 3],
|
| 88 |
["examples/9.mp4", "1.3B", 5, 3],
|
| 89 |
+
["examples/DJI_20250912163642_0003_D.mp4", "1.3B", 5, 3],
|
| 90 |
+
["examples/178db6e89ab682bfc612a3290fec58dd.mp4", "1.3B", 5, 3],
|
| 91 |
+
["examples/1b0daeb776471c7389b36cee53049417.mp4", "1.3B", 5, 3],
|
| 92 |
+
["examples/8a6dfb8cfe80634f4f77ae9aa830d075.mp4", "1.3B", 5, 3],
|
| 93 |
+
["examples/69230f105ad8740e08d743a8ee11c651.mp4", "1.3B", 5, 3],
|
| 94 |
+
["examples/b1f1fa44f414d7731cd7d77751093c44.mp4", "1.3B", 5, 3],
|
| 95 |
+
["examples/b68045aa2128ab63d9c7518f8d62eafe.mp4", "1.3B", 5, 3],
|
| 96 |
["examples/10.mp4", "1.3B", 5, 3],
|
| 97 |
["examples/30.mp4", "1.3B", 5, 3],
|
| 98 |
+
["examples/3.mp4", "1.3B", 5, 3],
|
| 99 |
["examples/31.mp4", "1.3B", 5, 3],
|
| 100 |
["examples/32.mp4", "1.3B", 5, 3],
|
| 101 |
+
|
| 102 |
["examples/35.mp4", "1.3B", 5, 3],
|
| 103 |
["examples/36.mp4", "1.3B", 5, 3],
|
| 104 |
["examples/39.mp4", "1.3B", 5, 3],
|
| 105 |
["examples/40.mp4", "1.3B", 5, 3],
|
| 106 |
+
["examples/2.mp4", "1.3B", 5, 3],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 107 |
["examples/DJI_20250912164311_0007_D.mp4", "1.3B", 5, 3],
|
| 108 |
]
|
| 109 |
|
|
|
|
| 273 |
raise ValueError(f"Unsupported model size: {model_size}")
|
| 274 |
|
| 275 |
|
| 276 |
+
|
|
|
|
|
|
|
|
|
|
| 277 |
|
| 278 |
|
| 279 |
|
|
|
|
| 426 |
|
| 427 |
|
| 428 |
# ==== 新增:上下翻转点云 ====
|
| 429 |
+
points[:, 2] = -points[:, 2]
|
| 430 |
# =========================
|
| 431 |
|
| 432 |
|