-
Notifications
You must be signed in to change notification settings - Fork 7k
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
ROI Pooling Layer #585
ROI Pooling Layer #585
Conversation
Hi @varunagrawal , Thanks a lot for the PR! |
@fmassa I took out the NMS and ROI Align layers in order to minimize the dependencies on this PR and tackle each thing separately. I don't know if this branch would merge with your |
When you create a PR, you can select to which branch you want it to send the PR to, see https://help.github.com/articles/creating-a-pull-request/ for some more information. |
Just saw your other PR. I think that's the way to go, I'd look into updating the layers branch.
|
@fmassa so do you want me to just base off the original Also, regarding that compilation error, it is a regression introduced by pytorch/pytorch#10301. Syed should have a fix in by next week. |
Oh yes, I had seen the diff but I didn't check the discussion in the PR. I think it's easier to base the diff off from the |
That should work. I'll force-update the PR to align with the |
Code for ROI Pooling layer.
(#477)