Daniellesry commited on
Commit
42437d6
·
1 Parent(s): 81d4d32
Files changed (1) hide show
  1. app.py +17 -18
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/2.mp4", "1.3B", 5, 3],
82
- ["examples/3.mp4", "1.3B", 5, 3],
83
- ["examples/5.mp4", "1.3B", 5, 3],
 
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
- ["examples/33.mp4", "1.3B", 5, 3],
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
- def flip_points(points):
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 = flip_points(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