Skip to content

Detectron for image/video region feature extraction, inspired by Xinlei's repo

Notifications You must be signed in to change notification settings

LuoweiZhou/detectron-vlp

Repository files navigation

Detectron for image/video feature extraction

Follow the official instructions to install Detectron (for inference only). This version of Detectron only supports python2 and has tested to support CUDA 8.0 and 9.0 at least. You can skip the steps on Caffe2 if you have torch installed (e.g., in the VLP or GVD conda env) and just finish the rest until here.

VLP

For VLP, download the corresponding config file and the checkpoint file and place under this root dir. Refer to extract_feat_flickr30k.sh and tools/extract_features.py for the usage.

GVD

For GVD, download the corresponding config file (rename to e2e_faster_rcnn_X-101-64x4d-FPN_2x-gvd.yaml) and the checkpoint file (rename to e2e_faster_rcnn_X-101-64x4d-FPN_2x-gvd.pkl) and place under this root dir. Refer to extract_feat_gvd_anet.sh and tools/extract_features_gvd_anet.py for the usage.

About

Detectron for image/video region feature extraction, inspired by Xinlei's repo

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published