Added setup.sh to install requirements

manogna-s · Jun 17, 2021 · 48c6516 · 48c6516
1 parent 750be5e
commit 48c6516
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/setup.sh b/setup.sh
@@ -0,0 +1,10 @@
+mkdir checkpoints
+mkdir checkpoints/pretrained_ckpts
+cd checkpoints/pretrained_ckpts/
+#wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz
+cd ../..
+mkdir checkpoints/output_ckpts
+pip install numpy tqdm torch torchvision scipy pillow matplotlib tabulate ml-collections tensorboard tensorflow
+cd cdfsl_dataset
+python3 setup.py install
+cd ..