SWA Object Detection for python
This project hosts the scripts for training SWA object detectors, as presented in our paper:
@article{zhang2020swa,
title={SWA Object Detection},
author={Zhang, Haoyang and Wang, Ying and Dayoub, Feras and S{"u}nderhauf, Niko},
journal={arXiv preprint arXiv:2012.12645},
year={2020}
}
The full paper is available at: https://arxiv.org/abs/2012.12645.
Introduction
Do you want to improve 1.0 AP for your object detector without any inference cost and any change to your detector? Let us tell you such a recipe. It is surprisingly simple: train your detector for an extra 12 epochs using cyclical learning rates and then average these 12 checkpoints as your final detection model. This potent recipe is inspired by Stochastic Weights Averaging (SWA), which is proposed in [1] for improving generalization in deep neural networks. We