Config MS_DETR for VOC0712 #17
Comments
|
Hi Khoa, Thank you for your interest in our work. Our implementation is based on Deformable-DETR, which sets the number of classes in COCO to 91 by default. The original COCO paper mentions 91 classes, but the commonly used dataset only includes 80 object detection classes, as explained in this reference. The extra classes do not affect the final performance since they are neither trained nor used, and the corresponding weights in the linear classifier are essentially ignored. Using classes in the range [1, 80] does not impact performance because the unused classes do not influence the training. Additionally, the class with ID 0 is reserved for foreground classification, where all objects are classified to class ID 0 in the encoder head, which is used in two-stage training. I hope this addresses your concerns. Please feel free to reach out if you have further questions. |
|
Really appreciate for your fast response. In addition, I just changed the num_classes to 21 and trained the MS-DETR on VOC0712 (16551 samples). Unfortunately, the accuracy is too low, could you give me some guidance on this? Below is the log after the first epoch and 34th epochs: After the first epochs: After 34th epochs: The accuracy on evaluation datasets after 34th epochs training: |
|
Dear Zhao Chuyang, I made some mistakes when trying to add some code for debugging. I can train on the VOC0712 right now, sorry for bothering you. |
Dear Authors,
I am currently configuring your model on the VOC0712 dataset (a combination of VOC 2007 and VOC 2012), which contains 20 object detection classes. I have converted the VOC annotations to the COCO annotation format using this link.
In the build(args) function (located in models/deformable_detr.py), I set the num_classes variable to 21. Although VOC0712 only has 20 classes with IDs ranging from 1 to 20, setting the variable to 20 causes a 'cuda/IndexKernel.cu:92 error.' As a workaround, I followed the suggestion in this link and increased the number of classes by 1 to ensure that the IDs remain within the range [0: num_classes-1]. And it frees the error.
After reviewing the annotations, I noticed that the category IDs for COCO 2017 are as follows: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 27, 28, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 67, 70, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 84, 85, 86, 87, 88, 89, 90], which corresponds to 80 classes.
Could you please clarify why you set num_classes to 91, despite there being only 80 categories? Additionally, could you explain why the category IDs are not converted to a range from 0 to 79 instead?
Thank you for your time and assistance.
Kind regards,
Khoa
The text was updated successfully, but these errors were encountered: