search

pzzhang / VinVL

project page for VinVL
351 stars 25 forks source link

Pre-exacted image features #10

Open stopmosk opened 3 years ago

stopmosk commented 3 years ago

Hi!

I'm trying to use VinVL model and scripts described in https://github.com/pzzhang/VinVL/blob/main/DOWNLOAD.md to extract image features from COCO, but I found that extracted COCO features (detections & labels) differ from pre-exacted image features.

Can you provide pretrained VinVL model and/or config (which is used to make pre-exacted features) for proper extraction?

rgtjf commented 3 years ago

+1

pzzhang commented 3 years ago

Please refer to the repo https://github.com/microsoft/scene_graph_benchmark for feature extraction.

If you only find minor difference in the predictions (by looking at the visualization tools/demo/demo_image.py), then the extracted features should be fine.