Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,13 +32,13 @@ from mono.utils.unproj_pcd import reconstruct_pcd, save_point_cloud
|
|
| 32 |
import gradio as gr
|
| 33 |
|
| 34 |
os.chdir(CODE_SPACE)
|
| 35 |
-
cfg = Config.fromfile('
|
| 36 |
|
| 37 |
torch.hub.download_url_to_file('https://images.unsplash.com/photo-1437622368342-7a3d73a34c8f', 'turtle.jpg')
|
| 38 |
torch.hub.download_url_to_file('https://images.unsplash.com/photo-1519066629447-267fffa62d4b', 'lions.jpg')
|
| 39 |
|
| 40 |
model = get_configured_monodepth_model(cfg, )
|
| 41 |
-
model, _, _, _ = load_ckpt(
|
| 42 |
model.eval()
|
| 43 |
|
| 44 |
device = "cpu"
|
|
|
|
| 32 |
import gradio as gr
|
| 33 |
|
| 34 |
os.chdir(CODE_SPACE)
|
| 35 |
+
cfg = Config.fromfile('./mono/configs/_base_/models/encoder_decoder/dino_vit_small_reg.dpt_raft.py')
|
| 36 |
|
| 37 |
torch.hub.download_url_to_file('https://images.unsplash.com/photo-1437622368342-7a3d73a34c8f', 'turtle.jpg')
|
| 38 |
torch.hub.download_url_to_file('https://images.unsplash.com/photo-1519066629447-267fffa62d4b', 'lions.jpg')
|
| 39 |
|
| 40 |
model = get_configured_monodepth_model(cfg, )
|
| 41 |
+
model, _, _, _ = load_ckpt('./weight/metric_depth_vit_small_800k.pth.pth', model, strict_match=False)
|
| 42 |
model.eval()
|
| 43 |
|
| 44 |
device = "cpu"
|