Skip to content

Commit

Permalink
update prepare
Browse files Browse the repository at this point in the history
  • Loading branch information
LinghaoChan committed Nov 18, 2024
1 parent 0b86ebc commit 34bd8ff
Show file tree
Hide file tree
Showing 4 changed files with 34 additions and 1 deletion.
1 change: 0 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,6 @@ Thumbs.db
**/*.pyc
*.pyc
*.out
*.sh
*.png
*.pth
*.pt
Expand Down
12 changes: 12 additions & 0 deletions prepare/download_glove.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
cd ./data/

echo -e "Downloading glove (in use by the evaluators)"
gdown --fuzzy https://drive.google.com/file/d/1cmXKUT31pqd7_XpJAiWEo1K81TMYHA5n/view?usp=sharing
rm -rf glove

unzip glove.zip
echo -e "Cleaning\n"
rm glove.zip
cd ..

echo -e "Downloading done!"
17 changes: 17 additions & 0 deletions prepare/download_t2m_evaluators.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
mkdir -p data/
cd data/
mkdir -p checkpoints/
cd checkpoints/

echo "The t2m evaluators will be stored in the './deps' folder"

echo "Downloading"
gdown --fuzzy https://drive.google.com/file/d/16hyR4XlEyksVyNVjhIWK684Lrm_7_pvX/view?usp=sharing
echo "Extracting"
unzip t2m.zip
echo "Cleaning"
rm t2m.zip

cd ../..

echo "Downloading done!"
5 changes: 5 additions & 0 deletions prepare/prepare_clip.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
mkdir -p deps/
cd deps/
git lfs install
git clone https://huggingface.co/openai/clip-vit-large-patch14
cd ..

0 comments on commit 34bd8ff

Please sign in to comment.