soo89 / ISD-SSD

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

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