https://github.com/ShoufaChen/DiffusionDet

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

Last synced: 15 days ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/ShoufaChen/DiffusionDet
• Owner: ShoufaChen
• License: other
• Created: 2022-11-17T04:34:54.000Z (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2022-12-22T04:40:55.000Z (almost 2 years ago)
• Last Synced: 2024-10-11T18:09:50.553Z (about 1 month ago)
• Language: Python
• Homepage:
• Size: 909 KB
• Stars: 2,073
• Watchers: 17
• Forks: 161
• Open Issues: 72
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
## DiffusionDet: Diffusion Model for Object Detection

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

![](teaser.png)

> [**DiffusionDet: Diffusion Model for Object Detection**](https://arxiv.org/abs/2211.09788)               

> [Shoufa Chen](https://www.shoufachen.com/), [Peize Sun](https://peizesun.github.io/), [Yibing Song](https://ybsong00.github.io/), [Ping Luo](http://luoping.me/)                 

> *[arXiv 2211.09788](https://arxiv.org/abs/2211.09788)* 

## Updates

- (11/2022) Code is released.

## Models

Method | Box AP (1 step) | Box AP (4 step) | Download

--- |:---:|:---:|:---:

[COCO-Res50](configs/diffdet.coco.res50.yaml) | 45.5 | 46.1 | [model](https://github.com/ShoufaChen/DiffusionDet/releases/download/v0.1/diffdet_coco_res50.pth)

[COCO-Res101](configs/diffdet.coco.res101.yaml) | 46.6 | 46.9 | [model](https://github.com/ShoufaChen/DiffusionDet/releases/download/v0.1/diffdet_coco_res101.pth)

[COCO-SwinBase](configs/diffdet.coco.swinbase.yaml) | 52.3 | 52.7 | [model](https://github.com/ShoufaChen/DiffusionDet/releases/download/v0.1/diffdet_coco_swinbase.pth)

[LVIS-Res50](configs/diffdet.lvis.res50.yaml) | 30.4 | 31.8 | [model](https://github.com/ShoufaChen/DiffusionDet/releases/download/v0.1/diffdet_lvis_res50.pth)

[LVIS-Res101](configs/diffdet.lvis.res101.yaml) | 31.9 | 32.9 | [model](https://github.com/ShoufaChen/DiffusionDet/releases/download/v0.1/diffdet_lvis_res101.pth)

[LVIS-SwinBase](configs/diffdet.lvis.swinbase.yaml) | 40.6 | 41.9 | [model](https://github.com/ShoufaChen/DiffusionDet/releases/download/v0.1/diffdet_lvis_swinbase.pth)

## Getting Started

The installation instruction and usage are in [Getting Started with DiffusionDet](GETTING_STARTED.md).

## License

This project is under the CC-BY-NC 4.0 license. See [LICENSE](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.

```BibTeX

@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}

}

```