search

ShoufaChen / DiffusionDet

[ICCV2023 Best Paper Finalist] PyTorch implementation of DiffusionDet (https://arxiv.org/abs/2211.09788)
Other
2.07k stars 161 forks source link

training on custom dataset #50

Open leo96code opened 1 year ago

leo96code commented 1 year ago

Really loving this breakthrough! Can you kindly explain the steps or make a Colab notebook for training DiffusionDet on a custom dataset? (maybe you can use the balloon dataset as used in detectron2). This will really help beginners like me. Thanks :)

ShoufaChen commented 1 year ago

Hi, Thank you for your feedback. I apologize for the delayed response due to the holiday season.

Could this issue https://github.com/ShoufaChen/DiffusionDet/issues/20 solve your question?