-
Notifications
You must be signed in to change notification settings - Fork 6.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
retinanet num_classes includes the background #8432
Comments
Sorry @ABCCASa , I don't understand the question.
What does that refer to? |
Hi, @NicolasHug , Thanks for your reply, |
Does this help: #4106 (comment) ?
|
@NicolasHug Yes, Thank you so much. |
📚 The doc issue
https://pytorch.org/vision/master/models/generated/torchvision.models.detection.retinanet_resnet50_fpn_v2.html
Hi, I am currently use the retinanet, the documentation says num_classes of retinanet includes the background, but when I look at the source code of RetinaNetClassificationHead, it use a tensor with all zeros to represent background. So do we still need to use index=0 as the background?
Suggest a potential alternative/fix
Maybe num_classes is exclude background?
The text was updated successfully, but these errors were encountered: