Skip to content
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

Potential Missing Filtering of -1 Labels in rcnn_cls_labels During Binary Cross-Entropy Loss Calculation in PointRCNN #1682

Open
ChunhengH opened this issue Nov 8, 2024 · 0 comments

Comments

@ChunhengH
Copy link

I was training pointRCNN and other point-based models using my own dataset and noticed that rcnn_cls_labels came up -1, which caused the training to terminate. Comparing to the original model I found that it seems that the -1 is not filtered when doing the binary cross entropy loss calculation. pointRCNN original model uses cls_valid_mask to filter all labels greater than 0, but in pcdet/models/roi_heads/roi_head_template.py it seems that not perform this step. I'm not sure if this is a problem.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant