diff --git a/static/images/teaser.png b/static/images/teaser.png index e0c6ada..09a42ea 100644 Binary files a/static/images/teaser.png and b/static/images/teaser.png differ diff --git a/training/download.py b/training/download.py new file mode 100644 index 0000000..19ec8bb --- /dev/null +++ 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 new file mode 100644 index 0000000..44de3f2 --- /dev/null +++ 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 \ No newline at end of file