Pytorch implementation of "AutoAssign: Differentiable Label Assignment for Dense Object Detection".

This is a PyTorch implementation of the AutoAssign paper:

  title={AutoAssign: Differentiable Label Assignment for Dense Object Detection},
  author={Zhu, Benjin and Wang, Jianfeng and Jiang, Zhengkai and Zong, Fuhang and Liu, Songtao and Li, Zeming and Sun, Jian},
  journal={arXiv preprint arXiv:2007.03496},

Get Started

  1. install cvpods following the instructions
# Install cvpods
git clone
cd cvpods 
## build cvpods (requires GPU)
pip install -r requirements.txt
python build develop
## preprare data path
mkdir datasets
ln -s /path/to/your/coco/dataset datasets/coco
  1. run the project
cd auto_assign.res50.fpn.coco.800size.1x

# train
pods_train --num-gpus 8

# test
pods_test --num-gpus 8
# test with provided weights
pods_test --num-gpus 8 MODEL.WEIGHTS /path/to/your/model.pth


Model Multi-scale training Multi-scale testing Testing time / im AP (minival) Link
AutoAssign_Res50_FPN_1x No No 53ms 40.5 download