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

How are cls_score and bbox_pred resized to match labels and bbox_targets #13

Open
AlexTS1980 opened this issue May 30, 2018 · 0 comments

Comments

@AlexTS1980
Copy link

In the training phase the number of predicted RoIs + scores is determined by hyperparameters like POST_NMS_ROIS_TRAINING. Caffe inits layers to (1,21) or (1,84) and then they are resized to match the number of predictions, e.g. (1000, 21) and (1000,84). This is done in custom layers, and I don't have a problem with that. At the same time, in order to get the loss, layers like cls_score, bbox_pred, fc6/7, are also resized to match the size (while keeping the number of weights the same). I don't see how Caffe does it out of the box, but I can't find the wrapper for that. Could anyone help?

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