download_data_and_checkpoints.sh

initial_dir=$(pwd)

mkdir -p data
mkdir -p checkpoints

# Download checkpoints
wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt -O checkpoints/v1-5-pruned-emaonly.ckpt &&
wget "https://www.dropbox.com/scl/fi/ox7tvyhsoqyhkkrf4z47b/tadp_depth_blipmin40.ckpt?rlkey=rocsl40cdia8mu28culdrjoo3&dl=1" -O checkpoints/tadp_depth_blipmin40.ckpt &&
wget "https://www.dropbox.com/scl/fi/men4sn4khyht5i1h6cdyh/tadp_seg_blipmin40.ckpt?rlkey=onlpos0js4g3wsm82ycku0rwd&dl=1" -O checkpoints/tadp_seg_blipmin40.ckpt &&
wget "https://www.dropbox.com/scl/fi/t93l3wcfkrvybfol4udes/tadp_watercolor.pt?rlkey=gjr4yhextamxck545etxe5rfh&dl=1" -O checkpoints/tadp_watercolor.pt &&
wget "https://www.dropbox.com/scl/fi/y6ah1z3wrig80op4g2w7x/tadp_nighttimeDriving.pth?rlkey=h0uzl69jvi3ulptvrxgjbwaoo&dl=1" -O checkpoints/tadp_nighttimeDriving.pth

# tokens
wget "https://www.dropbox.com/scl/fi/f9vema0ok1obeh69y2n7j/tokens.zip?rlkey=6fqy1rysmqznz238cisq3kgf2&dl=1" -O tokens.zip &&
unzip tokens.zip -d ./ && rm tokens.zip

# Perform actions based on the argument
if [ "$1" = "seg" ]; then
  echo "Downloading ADE20k"

cd data && mkdir ade && cd ade &&
wget http://data.csail.mit.edu/places/ADEchallenge/ADEChallengeData2016.zip && unzip ADEChallengeData2016.zip && rm ADEChallengeData2016.zip &&
wget http://data.csail.mit.edu/places/ADEchallenge/ADEChallengeData2016/annotations.zip && unzip annotations.zip && rm annotations.zip &&
wget http://data.csail.mit.edu/places/ADEchallenge/release_test.zip && unzip release_test.zip && rm release_test.zip &&
cd ../..
fi

if [ "$1" = "depth" ]; then
echo "Downloading nyu"
mkdir data/nyu_depth_v2 &&
wget http://horatio.cs.nyu.edu/mit/silberman/nyu_depth_v2/nyu_depth_v2_labeled.mat -O data/nyu_depth_v2/nyu_depth_v2_labeled.mat  &&
python datasets/depth/nyu_extract_official_train_test_set_from_mat.py data/nyu_depth_v2/nyu_depth_v2_labeled.mat datasets/depth/nyu_splits.mat data/nyu_depth_v2/official_splits/ &&
cd data/nyu_depth_v2 && mkdir sync && cd sync &&
gdown https://drive.google.com/uc?id=16JqgvqtICEsyi6dX85F6BB7AwI0qF9aP &&
unzip sync.zip && rm sync.zip
fi

if [ "$1" = "pascal_seg" ]; then
wget -P ./data/ http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
tar -xvf ./data/VOCtrainval_11-May-2012.tar -C ./data/
wget -O ./data/SegmentationClassAug.zip https://www.dropbox.com/s/oeu149j8qtbs1x0/SegmentationClassAug.zip?dl=0
unzip ./data/SegmentationClassAug.zip "SegmentationClassAug/*" -d ./data
rm -rf ./data/VOCdevkit/VOC2012/SegmentationClass
mv ./data/SegmentationClassAug ./data/VOCdevkit/VOC2012/SegmentationClass
cp datasets/trainaug.txt ./data/VOCdevkit/VOC2012/ImageSets/Segmentation/
fi


if [ "$1" = "cross_cityscapes" ]; then
# download and put cityscapes zips in
if [ -d /data/cityscapes/ ]; then
    echo "Cannot find data/cityscapes directory. Make sure you have downloaded the data and rerun the script."
else
  echo "Found Cityscapes data. Extracting..."
  mv ./data/cityscapes/gtFine_trainvaltest/gtFine ./data/cityscapes/
  mv ./data/cityscapes/leftImg8bit_trainvaltest/leftImg8bit ./data/cityscapes/
  mv ./data/cityscapes/leftImg8bit_trainextra/leftImg8bit ./data/cityscapes/
  rm -rf ./data/cityscapes/gtFine_trainvaltest
  rm -rf ./data/cityscapes/leftImg8bit_trainvaltest
  rm -rf ./data/cityscapes/leftImg8bit_trainextra
  git clone https://github.com/open-mmlab/mmsegmentation.git
  python -m pip install cityscapesscripts
  mim install mmengine
  python mmsegmentation/tools/dataset_converters/cityscapes.py data/cityscapes --nproc 8
fi
  cd data &&
  mkdir dark_zurich && cd dark_zurich &&
  wget https://data.vision.ee.ethz.ch/csakarid/shared/GCMA_UIoU/Dark_Zurich_val_anon.zip && unzip Dark_Zurich_val_anon.zip && rm Dark_Zurich_val_anon.zip
  cd ../../

  cd data && mkdir NighttimeDrivingTest && cd NighttimeDrivingTest
  wget http://data.vision.ee.ethz.ch/daid/NighttimeDriving/NighttimeDrivingTest.zip && unzip NighttimeDrivingTest.zip && rm NighttimeDrivingTest.zip
  cd ../../
fi


if [ "$1" = "cross_pascal" ]; then
wget -P ./data/ http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
tar -xvf ./data/VOCtrainval_11-May-2012.tar -C ./data/

wget -O ./data/SegmentationClassAug.zip https://www.dropbox.com/s/oeu149j8qtbs1x0/SegmentationClassAug.zip?dl=0
unzip ./data/SegmentationClassAug.zip "SegmentationClassAug/*" -d ./data

rm -rf ./data/VOCdevkit/VOC2012/SegmentationClass
mv ./data/SegmentationClassAug ./data/VOCdevkit/VOC2012/SegmentationClass

cp datasets/trainaug.txt ./data/VOCdevkit/VOC2012/ImageSets/Segmentation/

git clone https://github.com/naoto0804/cross-domain-detection.git data/cross-domain-detection
bash data/cross-domain-detection/datasets/prepare.sh
mv watercolor data/cross-domain-detection/datasets/
mv comic data/cross-domain-detection/datasets/
mv clipart data/cross-domain-detection/datasets/
fi

cd "$initial_dir"
### setup da former stuff