search

fundamentalvision / Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.
Apache License 2.0
3.23k stars 520 forks source link

How use the pretrained models for custom dataset #117

Open ver0z opened 2 years ago

ver0z commented 2 years ago

As in the main DETR there is a way of using the models that were pretrained on coco by deleting the heads for use that on a specific dataset (custom), I tried to make the same here but it gives an error that says is missing keys. So is there a way of use the pretrained models available here? I know I can do the approach of train from scratch, but as an alternative way for that.

daimakram commented 2 years ago

Hi @ver0z, were to able to find out which layers to delete? Will appreciate your response.