bottom-up-attention.pytorch

环境

torch1.5+cu 101+cp38 on 2080ti

# clone the repository inclduing Detectron2(@be792b9) 
$ git clone --recursive https://github.com/MILVLG/bottom-up-attention.pytorch

$ cd detectron2
$ pip install -e .
$ cd ..

detectron2直接克隆有问题，需要把detecron2/detecron2中的所有文件放入detecron2下

git clone https://github.com/ptrblck/apex.git
cd apex
pip install -v --no-cache-dir ./


apt-get install libglib2.0-dev

 Faster R-CNN-k10-100 is the default model地址https://awma1-my.sharepoint.com/:u:/g/personal/yuz_l0_tn/EaXvCC3WjtlLvvEfLr3oa8UBLA21tcLh4L8YLbYXl6jgjg?download=1

python3 extract_features.py --mode caffe \
         --num-cpus 0 --gpus 1 \
         --extract-mode roi_feats \
         --min-max-boxes '36' \
         --config-file configs/caffe/test-caffe-r101.yaml \
         --image-dir /hy-tmp/img --bbox-dir /hy-tmp/out --out-dir /hy-tmp/out