Spaces:
Running
Running
Update app
Browse files
app.py
CHANGED
|
@@ -32,8 +32,8 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
|
|
| 32 |
# depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
|
| 33 |
depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small", device=0)
|
| 34 |
|
| 35 |
-
total_params = sum(param.numel() for param in depth_anything.parameters())
|
| 36 |
-
print('Total parameters: {:.2f}M'.format(total_params / 1e6))
|
| 37 |
|
| 38 |
transform = Compose([
|
| 39 |
Resize(
|
|
|
|
| 32 |
# depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
|
| 33 |
depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small", device=0)
|
| 34 |
|
| 35 |
+
# total_params = sum(param.numel() for param in depth_anything.parameters())
|
| 36 |
+
# print('Total parameters: {:.2f}M'.format(total_params / 1e6))
|
| 37 |
|
| 38 |
transform = Compose([
|
| 39 |
Resize(
|