Spaces:
Runtime error
Runtime error
Roman Bachmann
commited on
Commit
·
1ae6821
1
Parent(s):
051d843
fix omnidata dl
Browse files
app.py
CHANGED
@@ -75,11 +75,7 @@ def plot_semseg(img, semseg, ax):
|
|
75 |
|
76 |
|
77 |
# Initialize Omnidata depth model
|
78 |
-
|
79 |
-
os.system("wget https://datasets.epfl.ch/vilab/iccv21/weights/omnidata_rgb2depth_dpt_hybrid.pth")
|
80 |
-
os.system("unzip -j download -d pretrained_models")
|
81 |
-
os.system("rm download")
|
82 |
-
|
83 |
omnidata_ckpt = torch.load('./pretrained_models/omnidata_rgb2depth_dpt_hybrid.pth', map_location='cpu')
|
84 |
depth_model = DPTDepthModel()
|
85 |
depth_model.load_state_dict(omnidata_ckpt)
|
|
|
75 |
|
76 |
|
77 |
# Initialize Omnidata depth model
|
78 |
+
os.system("wget https://datasets.epfl.ch/vilab/iccv21/weights/omnidata_rgb2depth_dpt_hybrid.pth -P pretrained_models")
|
|
|
|
|
|
|
|
|
79 |
omnidata_ckpt = torch.load('./pretrained_models/omnidata_rgb2depth_dpt_hybrid.pth', map_location='cpu')
|
80 |
depth_model = DPTDepthModel()
|
81 |
depth_model.load_state_dict(omnidata_ckpt)
|