update

TIGER-AI-Lab · Jun 24, 2024 · 5f0ef0a · 5f0ef0a
1 parent c5a1b3e
commit 5f0ef0a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/static/images/teaser.png b/static/images/teaser.png
diff --git a/training/download.py b/training/download.py
@@ -0,0 +1,4 @@
+from huggingface_hub import hf_hub_download
+
+hf_hub_download(repo_id="TIGER-Lab/VideoFeedback", filename="annotated/frames_annotated_train.zip", repo_type="dataset", local_dir="./data")
+hf_hub_download(repo_id="TIGER-Lab/VideoFeedback", filename="annotated/frames_real_train.zip", repo_type="dataset", local_dir="./data")
diff --git a/training/download.sh b/training/download.sh
@@ -0,0 +1,11 @@
+
+mkdir -p ./data
+
+if [ -d "${bench_name}/${split}/frames_${postfix}" ]; then
+    echo "frames exists"
+else
+    echo "frames not exist, downloading..."
+    wget "https://huggingface.co/datasets/TIGER-Lab/VideoFeedback/resolve/main/real/frames_real_train.zip" -O "./data/frames_real_train.zip"
+    unzip "./data/frames_real_train.zip" -d "./data/images/"
+    rm "./data/frames_real_train.zip"    
+fi