Zenodo hosting

omnidata_tools/torch/tools/download_surface_normal_models.sh

@@ -1,19 +1,25 @@
 ##!/usr/bin/env bash
 
+# Switch hosting 
 # wget https://drive.switch.ch/index.php/s/RFfTZwyKROKKx0l/download
 # unzip -j download -d pretrained_models
 # rm download
 
-echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" | sudo tee -a /etc/apt/sources.list.d/google-cloud-sdk.list
-sudo apt-get install apt-transport-https ca-certificates gnupg
-curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | sudo apt-key --keyring /usr/share/keyrings/cloud.google.gpg add -
-sudo apt-get update -y && sudo apt-get install -y google-cloud-sdk
+# Gdrive hosting
+# echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" | sudo tee -a /etc/apt/sources.list.d/google-cloud-sdk.list
+# sudo apt-get install apt-transport-https ca-certificates gnupg
+# curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | sudo apt-key --keyring /usr/share/keyrings/cloud.google.gpg add -
+# sudo apt-get update -y && sudo apt-get install -y google-cloud-sdk
 
-sudo apt install -y imagemagick
+# sudo apt install -y imagemagick
 
-pip install gdown
-mkdir -p pretrained_models 
+# pip install gdown
+# mkdir -p pretrained_models 
 
 # https://drive.google.com/uc?id=1wNxVO4vVbDEMEpnAi_jwQObf2MFodcBR
-gdown '1iJjV9rkdeLvsTU9x3Vx8vwZUg-sSQ9nm&confirm=t' -O ./pretrained_models/ # omnidata normals (v1)
-gdown '1wNxVO4vVbDEMEpnAi_jwQObf2MFodcBR&confirm=t' -O ./pretrained_models/ # omnidata normals (v2)
+# gdown '1iJjV9rkdeLvsTU9x3Vx8vwZUg-sSQ9nm&confirm=t' -O ./pretrained_models/ # omnidata normals (v1)
+# gdown '1wNxVO4vVbDEMEpnAi_jwQObf2MFodcBR&confirm=t' -O ./pretrained_models/ # omnidata normals (v2)
+
+mkdir -p pretrained_models && cd pretrained_models
+wget 'https://zenodo.org/records/10447888/files/omnidata_dpt_normal_v2.ckpt?download=1'
+wget 'https://zenodo.org/records/10447888/files/omnidata_unet_normal_v1.pth?download=1'