ShoufaChen/DiffusionDet

[ICCV2023 Best Paper Finalist] PyTorch implementation of DiffusionDet (https://arxiv.org/abs/2211.09788)

Python

2,259

6 commits

updated Dec 22, 2022

See the code

README

DiffusionDet: Diffusion Model for Object Detection

DiffusionDet is the first work of diffusion model for object detection.

DiffusionDet: Diffusion Model for Object Detection
Shoufa Chen, Peize Sun, Yibing Song, Ping Luo
arXiv 2211.09788

Updates

  • (11/2022) Code is released.

Models

MethodBox AP (1 step)Box AP (4 step)Download
COCO-Res5045.546.1model
COCO-Res10146.646.9model
COCO-SwinBase52.352.7model
LVIS-Res5030.431.8model
LVIS-Res10131.932.9model
LVIS-SwinBase40.641.9model

Getting Started

The installation instruction and usage are in Getting Started with DiffusionDet.

License

This project is under the CC-BY-NC 4.0 license. See LICENSE for details.

Citing DiffusionDet

If you use DiffusionDet in your research or wish to refer to the baseline results published here, please use the following BibTeX entry.

@article{chen2022diffusiondet,
      title={DiffusionDet: Diffusion Model for Object Detection},
      author={Chen, Shoufa and Sun, Peize and Song, Yibing and Luo, Ping},
      journal={arXiv preprint arXiv:2211.09788},
      year={2022}
}

ShoufaChen/DiffusionDet

[ICCV2023 Best Paper Finalist] PyTorch implementation of DiffusionDet (https://arxiv.org/abs/2211.09788)

Python

2,259

6 commits

updated Dec 22, 2022

See the code

README

DiffusionDet: Diffusion Model for Object Detection

DiffusionDet is the first work of diffusion model for object detection.

DiffusionDet: Diffusion Model for Object Detection
Shoufa Chen, Peize Sun, Yibing Song, Ping Luo
arXiv 2211.09788

Updates

  • (11/2022) Code is released.

Models

MethodBox AP (1 step)Box AP (4 step)Download
COCO-Res5045.546.1model
COCO-Res10146.646.9model
COCO-SwinBase52.352.7model
LVIS-Res5030.431.8model
LVIS-Res10131.932.9model
LVIS-SwinBase40.641.9model

Getting Started

The installation instruction and usage are in Getting Started with DiffusionDet.

License

This project is under the CC-BY-NC 4.0 license. See LICENSE for details.

Citing DiffusionDet

If you use DiffusionDet in your research or wish to refer to the baseline results published here, please use the following BibTeX entry.

@article{chen2022diffusiondet,
      title={DiffusionDet: Diffusion Model for Object Detection},
      author={Chen, Shoufa and Sun, Peize and Song, Yibing and Luo, Ping},
      journal={arXiv preprint arXiv:2211.09788},
      year={2022}
}

Significant stargazers

Jitesh Jain

127 followers · starred Nov 2022

Yuxin Wu

4,004 followers · starred Dec 2022

Zhang Li

31 followers · starred Feb 2024

Kunchang Li

385 followers · starred Nov 2022