CCOP
Code of our paper Contrastive Object-level Pre-training with Spatial Noise Curriculum Learning
Requirement
- Install OpenSelfSup
- Install Detectron2, Do not forget to setup Detectron2 datasets!!!
- Install Kornia for fast data augmentation
Usage
Run Selective Search
% remember to setup the dataset paths
python tools/selective_search.py
Setup dataset
mkdir data
ln -s path_to_coco data
Run CCOP pre-training and Mask R-CNN benchmark
% training a ResNet-50 model with 8 GPU
zsh tools/det_train_benchmark.sh configs/selfsup/ccop/r50_d2.py 8 path_to_output
Citation
@article{yang2021contrastive,
title={Contrastive Object-level Pre-training with Spatial Noise Curriculum Learning},
author={Yang, Chenhongyi and Huang, Lichao and Crowley, Elliot J},
journal={arXiv preprint arXiv:2111.13651},
year={2021}
}