Custom Dataset

[SayBender]

Thanks for the amazing work.

I like to know your recommendations on how to train on custom dataset with lower number of possible classes (say 3), Then the number of queries would also need be changed. I have tired this with Deformable DETR and it does not let me change number of queries even after removing the classification head. I wonder if you fixed this issue.

Any recommendation on what and how to change the code for custom dataset implementation is very appreciated.