Skip to content
code for C-MIL: Continuation Multiple Instance Learning for Weakly Supervised Object Detection
Lua Cuda C CMake Shell
Branch: master
Clone or download
Latest commit 7f01ef8 Nov 27, 2018

README.md

C-MIL

code for C-MIL: Continuation Multiple Instance Learning for Weakly Supervised Object Detection

Environments

  • Ubuntu 16.04 LTS
  • NVIDIA V100 + CUDA9.0 + CuDNN7.0
  • Torch7

Detection Samples

Train and Test

  1. Install the dependencies

    cd ./C-MIL
    export DIR=$(pwd) 
    
    luarocks install hdf5 matio protobuf rapidjson loadcaffe xml
    
    cd $DIR/libs/functions
    sh install.sh
    
    cd $DIR/layers
    luarocks make
  2. Download dataset, proposals and ImageNet pre-trained model

    Download VOC2007 from: http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar

    Download proposals from: https://dl.dropboxusercontent.com/s/orrt7o6bp6ae0tc/selective_search_data.tgz

    Download VGGF from: http://www.robots.ox.ac.uk/~vgg/software/deep_eval/releases/bvlc/VGG_CNN_F.caffemodel https://gist.githubusercontent.com/ksimonyan/a32c9063ec8e1118221a/raw/6a3b8af023bae65669a4ceccd7331a5e7767aa4e/VGG_CNN_F_deploy.prototxt

    mkdir $DIR/data    
    mkdir $DIR/output

    The data folder has the following structure:

    $C-MIL/data/datasets/VOCdevkit_2007/
    $C-MIL/data/datasets/VOCdevkit_2007/VOCcode
    $C-MIL/data/datasets/VOCdevkit_2007/VOC2007
    $C-MIL/data/datasets/VOCdevkit_2007/...
    $C-MIL/data/datasets/proposals/
    $C-MIL/data/models/
    $C-MIL/data/results/
  3. Train, test and evaluate

    cd $DIR
    # train
    th train_cmil.lua 0 SSW
    # test
    th test_cmil.lua 0 SSW
    # evaluate
    th detection_mAP.lua 0 SSW output/path/to/scorefiles/score_test_epoch20.h5 2

Acknowledgements

Acknowledgements will be added later.

You can’t perform that action at this time.