IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.15k stars 232 forks source link

Train Dino with focalNet backbone on custom datasets #156

Closed 4BD0U closed 1 year ago

4BD0U commented 1 year ago

I am having difficulty training a Dino model with a FocalNet backbone on a custom dataset. Although I was able to successfully train the model using a SWIN-L backbone, after 10 epochs with FocalNet, all detection scores are still below 1%. I am unable to understand why this might be the case as the dataset used is identical and the pre-trained model provided in the GitHub repository was used as a starting point. Can you help me understand what might be causing this issue, please

Hezey commented 1 year ago

@4BD0U Hello, how to change backbone with others?Can you tell me ?please

HaoZhang534 commented 1 year ago

I am having difficulty training a Dino model with a FocalNet backbone on a custom dataset. Although I was able to successfully train the model using a SWIN-L backbone, after 10 epochs with FocalNet, all detection scores are still below 1%. I am unable to understand why this might be the case as the dataset used is identical and the pre-trained model provided in the GitHub repository was used as a starting point. Can you help me understand what might be causing this issue, please

I recommend to refer to this repo https://github.com/FocalNet/FocalNet-DINO .

4BD0U commented 1 year ago

Thank you for your response. The problem was that the pre-trained model provided was meant for Dino 5scale with focalNet, while the configuration file was set up for Dino 4scale.

4BD0U commented 1 year ago

I simply used the implemented backbone and set up the configuration file.