Spaces:
Runtime error
Runtime error
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"
|