File tree Expand file tree Collapse file tree 1 file changed +12
-2
lines changed
Expand file tree Collapse file tree 1 file changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -148,8 +148,18 @@ bash tools/scripts/download_pretrained_models.sh
148148
149149Before training, you will need to compute the BLIP embeddings for the videos/images. To do so, run:
150150``` bash
151- python tools/embs/save_blip_embs_vids.py # This will compute the embeddings for the WebVid-CoVR videos.
152- python tools/embs/save_blip_embs_imgs.py # This will compute the embeddings for the CIRR or FashionIQ images.
151+ # This will compute the embeddings for the WebVid-CoVR videos.
152+ # Note that you can use multiple GPUs with --num_shards and --shard_id
153+ python tools/embs/save_blip_embs_vids.py --video_dir datasets/WebVid/2M/train --todo_ids annotation/webvid-covr/webvid2m-covr_train.csv
154+
155+ # This will compute the embeddings for the WebVid-CoVR-Test videos.
156+ python tools/embs/save_blip_embs_vids.py --video_dir datasets/WebVid/8M/train --todo_ids annotation/webvid-covr/webvid8m-covr_test.csv
157+
158+ # This will compute the embeddings for the CIRR images.
159+ python tools/embs/save_blip_embs_imgs.py --image_dir datasets/CIRR/images/
160+
161+ # This will compute the embeddings for FashionIQ images.
162+ python tools/embs/save_blip_embs_imgs.py --image_dir datasets/fashion-iq/images/
153163```
154164
155165&emsp ;
You can’t perform that action at this time.
0 commit comments