Roman Bachmann commited on
Commit
1ae6821
·
1 Parent(s): 051d843

fix omnidata dl

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -75,11 +75,7 @@ def plot_semseg(img, semseg, ax):
75
 
76
 
77
  # Initialize Omnidata depth model
78
- #os.system("wget https://drive.switch.ch/index.php/s/RFfTZwyKROKKx0l/download")
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)