Skip to content

Commit 1b2d88a

Browse files
committed
Improve Computing BLIP embeddings section
1 parent 6ec5c54 commit 1b2d88a

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

README.md

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,8 +148,18 @@ bash tools/scripts/download_pretrained_models.sh
148148

149149
Before 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
 

0 commit comments

Comments
 (0)