search

ShoufaChen / DiffusionDet

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

An issue about training the Net in VOC #75

Closed sixone-Jiang closed 1 year ago

sixone-Jiang commented 1 year ago

Thank you for your brilliant work! I'm really sorry to disturb your time. I have two Questions as follows: First, I want to test your code in COCO, but I am really a freshman; how can I trained this model in VOC2012 with json annotations? Second, I try to frozen your backbone Net, and then fine-turn your sparse head; but I don't know how to frozen it right.

sixone-Jiang commented 1 year ago

I also read the #20, but I can`t fix my problem.

ShoufaChen commented 1 year ago

Hi,

Detectron2 supports the VOC dataset. You can refer https://github.com/facebookresearch/detectron2/blob/main/configs/PascalVOC-Detection/faster_rcnn_R_50_FPN.yaml for your experiment.

sixone-Jiang commented 1 year ago

Thanks！