Fix minor issues for final release

verlab · Apr 1, 2022 · 4e9782b · 4e9782b
1 parent 9db83c1
commit 4e9782b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -138,8 +138,12 @@ If you want to download the code and run it by yourself in your environment, or
     tar -xf rl_fast_forward/resources/YouCook2/scripts.tar.gz -C rl_fast_forward/resources/YouCook2/
     rm rl_fast_forward/resources/YouCook2/scripts.tar.gz
 
+    wget -O rl_fast_forward/resources/YouCook2/splits.tar.gz http://youcook2.eecs.umich.edu/static/YouCookII/splits.tar.gz
+    tar -xf rl_fast_forward/resources/YouCook2/splits.tar.gz -C rl_fast_forward/resources/YouCook2/
+    rm rl_fast_forward/resources/YouCook2/splits.tar.gz
+
     # Install youtube-dl and download the available videos
-    pip install youtube_dl
+    pip install youtube_dl # PS.: The YouTube-DL have been slow lately. If your download speed is under 100KiB/s, consider changing it to the YT-DLP fork (https://github.com/yt-dlp/yt-dlp)
     cd rl_fast_forward/resources/YouCook2/scripts
     python download_youcookii_videos.py
     ```

diff --git a/demos/extract_VDAN+_feats.ipynb b/demos/extract_VDAN+_feats.ipynb
@@ -89,7 +89,7 @@
         "\n",
         "# Run the code\n",
         "%cd semantic_encoding/_utils\n",
-        "! python extract_vdan_plus_feats.py -m ../models/vdan+_pretrained_model.pth -i ../resources/example.mp4 -u ../resources/example.txt -o ../resources/"
+        "! python extract_vdan_plus_feats.py -m ../models/vdan+_pretrained_model.pth -i <VIDEO_FILENAME> -u <DOCUMENT_FILENAME> -o <OUTPUT_FOLDER>"
       ]
     }
   ],

diff --git a/demos/train_SAFFA.ipynb b/demos/train_SAFFA.ipynb
@@ -80,6 +80,7 @@
       "source": [
         "# Download the pretrained model\n",
         "%cd /content/TextDrivenVideoAcceleration_TPAMI_2022\n",
+        "! mkdir semantic_encoding/models/\n",
         "! wget -O semantic_encoding/models/vdan+_pretrained_model.pth https://github.com/verlab/TextDrivenVideoAcceleration_TPAMI_2022/releases/download/pre_release/vdan+_pretrained_model.pth"
       ]
     },
@@ -125,11 +126,16 @@
       "outputs": [],
       "source": [
         "# 2. Download the scripts used to collect the videos + Install youtube-dl and download the available videos\n",
+        "%cd /content/TextDrivenVideoAcceleration_TPAMI_2022\n",
         "! wget -O rl_fast_forward/resources/YouCook2/scripts.tar.gz http://youcook2.eecs.umich.edu/static/YouCookII/scripts.tar.gz\n",
         "! tar -xf rl_fast_forward/resources/YouCook2/scripts.tar.gz -C rl_fast_forward/resources/YouCook2/\n",
         "! rm rl_fast_forward/resources/YouCook2/scripts.tar.gz\n",
         "\n",
-        "! pip install youtube_dl\n",
+        "! wget -O rl_fast_forward/resources/YouCook2/splits.tar.gz http://youcook2.eecs.umich.edu/static/YouCookII/splits.tar.gz\n",
+        "! tar -xf rl_fast_forward/resources/YouCook2/splits.tar.gz -C rl_fast_forward/resources/YouCook2/\n",
+        "! rm rl_fast_forward/resources/YouCook2/splits.tar.gz\n",
+        "\n",
+        "! pip install youtube_dl # PS.: The YouTube-DL have been slow lately. If your download speed is under 100KiB/s, consider changing it to the YT-DLP fork (https://github.com/yt-dlp/yt-dlp)\n",
         "%cd rl_fast_forward/resources/YouCook2/scripts\n",
         "! python download_youcookii_videos.py"
       ]