ISD: Interpolation-based Semi-supervised learning for object Detection (CVPR 2021)

March 9, 2021 ยท View on GitHub

By Jisoo Jeong, Vikas Verma, Minsung Hyun, Juho Kannala, Nojun Kwak

For more details, please refer to our arXiv paper

Installation & Preparation

We experimented with ISD using the SSD pytorch framework. To use our model, complete the installation & preparation on the SSD pytorch homepage

prerequisites

  • Python 3.6
  • Pytorch 1.5.0

Supervised learning

python train_ssd.py

CSD training

python train_csd.py

ISD training

python train_isd.py

Evaluation

python eval.py