Skip to content
Towards Human-Machine Cooperation: Self-supervised Sample Mining for Object Detection
Branch: master
Clone or download
Latest commit 5fb6810 Mar 26, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
caffe update code May 29, 2018
data update code May 29, 2018
experiments update code May 29, 2018
lib set al May 30, 2018
models update code May 29, 2018
tools remove double else Mar 21, 2019
LICENSE update code May 29, 2018
README.md Update README.md Mar 26, 2019
test.sh update code May 29, 2018
train.sh update code May 29, 2018

README.md

SSM

Towards Human-Machine Cooperation: Self-supervised Sample Mining for Object Detection

Keze Wang, Xiaopeng Yan, Dongyu Zhang, Lei Zhang, Liang Lin

Sun Yat-Sen University, Presented at CVPR2018

License

For Academic Research Use Only!

Citing SSM

If you find SSM useful in your research, please consider citing:

@InProceedings{Wang_2018_CVPR,
          author = {Wang, Keze and Yan, Xiaopeng and Zhang, Dongyu and Zhang, Lei and Lin, Liang},
          title = {Towards Human-Machine Cooperation: Self-Supervised Sample Mining for Object Detection},
          booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
          month = {June},
          year = {2018}
        }

Dependencies

The code is built on top of R-FCN. Please carefully read through py-R-FCN and make sure py-R-FCN can run within your enviornment.

Datasets/Pre-trained model

  1. In our paper, we used Pascal VOC2007/VOC2012 and COCO as our datasets, and ResNet-101 model as our pre-trained model.

  2. Please download ImageNet-pre-trained ResNet-101 model manually, and put them into $SSM_ROOT/data/imagenet_models

Usage

  1. training

    Before training, please prepare your dataset and pre-trained model and store them in the right path as R-FCN. You can go to ./tools/ and modify train_net.py to reset some parameters.Then, simply run sh ./train.sh.

  2. testing

    Before testing, you can modify test.sh to choose the trained model path, then simply run sh ./test.sh to get the evaluation result.

Misc

Tested on Ubuntu 14.04 with a Titan X GPU (12G) and Intel(R) Xeon(R) CPU E5-2623 v3 @ 3.00GHz.

Acknowledgement

Thanks for the contribution of Xiaoxi Wang.

You can’t perform that action at this time.