Alessio Grancini
commited on
Update monocular_depth_estimator.py
Browse files
monocular_depth_estimator.py
CHANGED
|
@@ -151,9 +151,8 @@ class MonocularDepthEstimator:
|
|
| 151 |
cv2.destroyAllWindows()
|
| 152 |
|
| 153 |
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
depth_estimator.run("assets/videos/testvideo2.mp4")
|
| 158 |
|
| 159 |
|
|
|
|
| 151 |
cv2.destroyAllWindows()
|
| 152 |
|
| 153 |
|
| 154 |
+
if __name__ == "__main__":
|
| 155 |
+
depth_estimator = MonocularDepthEstimator(model_type="dpt_hybrid_384")
|
| 156 |
+
depth_estimator.run("assets/videos/testvideo2.mp4")
|
|
|
|
| 157 |
|
| 158 |
|