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

Fixed Omnidata depth DPT download link

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -75,7 +75,8 @@ 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("unzip -j download -d pretrained_models")
80
  os.system("rm download")
81
 
 
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