Improved diffusion object detection through deformable sigmoid variance and adjustable sampling strategy
摘要
We propose improved DiffusionDet, an enhanced framework that integrates diffusion models into object detection. By introducing a deformable sigmoid variance schedule and an adjustable sampling strategy, improved DiffusionDet achieves improved sampling efficiency and accuracy, demonstrated by a 6.8AP increase in the VOC dataset using 500 initial boxes and 50 timesteps. Our framework effectively captures sparse features, addressing limitations in low-ground-truth density scenarios. The improved sampling strategy makes our detect network superior ability to sampling small object boxes on image boundaries. This work addresses the disadvantage of over-reliance on random coverage when sampling detect boxes in DiffusionDet as well as contributing to the development of more robust and efficient diffusion models for object detection. The code is available at: https://github.com/tengfeifun/diffusiondet-v2.