This project provides an implementation for our ECCV2022 paper "DenseTeacher: Dense Pseudo-Label for Semi-supervised Object Detection" on PyTorch.
- install cvpods locally (requires cuda to compile)
python3 -m pip install 'git+https://github.com/Megvii-BaseDetection/cvpods.git'
# (add --user if you don't have permission)
# Or, to install it from a local clone:
git clone https://github.com/Megvii-BaseDetection/cvpods.git
python3 -m pip install -e cvpods
# Or,
pip install -r requirements.txt
python3 setup.py build develop
- prepare datasets
cd /path/to/cvpods/datasets
ln -s /path/to/your/coco/dataset coco
- start training
cd DenseTeacher/coco-p10
pods_train --dir .
# Evaluation will be automatically start after each epoch
This repo is developed based on cvpods. Please check cvpods for more details and features.
This repo is released under the Apache 2.0 license. Please see the LICENSE file for more information.