Skip to content

hwanheelee1993/BUTD-UNITER-NLVR2

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

60 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This fork of bottom-up-attention is used to support feature extraction of faster-rcnn, which can be used through the pre-built Docker image chenrocks/butd-caffe:nlvr2.

Running Docker

bash scripts/extract_imgfeat.sh $PATH_TO_IMG_FOLDER $PATH_TO_IMG_NPY

Extracting Feature

Modify line 42-43 in 'tools/generate_npz.py' as follows.
line 42 MIN_BOXES = 100 -> MIN_BOXES = 36
line 43 MAX_BOXES = 100 -> MAX_BOXES = 36

```
python tools/generate_npz.py --def models/vg/ResNet-101/faster_rcnn_end2end_final/test.prototxt --cfg experiments/cfgs/faster_rcnn_end2end_resnet.yml --out /output/ --net 
data/faster_rcnn_models/resnet101_faster_rcnn_final.caffemodel --prefix butd

```
The output files will be stored in $PATH_TO_IMG_NPY as follows.
sample.jpg -> butd_sample.npz
(i.e. imgname.jpg -> butd_imgname.npz)

About

Support extracting BUTD features for NLVR2 images.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 63.9%
  • C++ 25.3%
  • Python 6.5%
  • Cuda 2.1%
  • CMake 0.9%
  • Shell 0.5%
  • Other 0.8%