Skip to content

Fix RoIAlignOp GPU implementation for RoIs without batch index#9230

Closed
malfet wants to merge 1 commit intopytorch:masterfrom
malfet:fix_roi_align_on_gpu
Closed

Fix RoIAlignOp GPU implementation for RoIs without batch index#9230
malfet wants to merge 1 commit intopytorch:masterfrom
malfet:fix_roi_align_on_gpu

Conversation

@malfet
Copy link
Contributor

@malfet malfet commented Jul 7, 2018

Summary: Fix RoIAlignOp GPU implementation for RoIs without batch index
According to https://caffe2.ai/docs/operators-catalogue.html#roialign, RoIs is "2D input of shape (R, 4 or 5)"
Pass RoIs 2nd dimension as kernel parameter and adjust kernel accordingly

Summary: Fix RoIAlignOp GPU implementation for RoIs without batch index
According to https://caffe2.ai/docs/operators-catalogue.html#roialign, RoIs is "2D input of shape (R, 4 or 5)"
Pass RoIs 2nd dimension as kernel parameter and adjust kernel accordingly
Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@malfet has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

Copy link
Member

@houseroad houseroad left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

jramseyer pushed a commit to jramseyer/pytorch that referenced this pull request Jul 30, 2018
…ch#9230)

Summary:
Fix RoIAlignOp GPU implementation for RoIs without batch index
According to https://caffe2.ai/docs/operators-catalogue.html#roialign, RoIs is "2D input of shape (R, 4 or 5)"
Pass RoIs 2nd dimension as kernel parameter and adjust kernel accordingly
Pull Request resolved: pytorch#9230

Reviewed By: houseroad

Differential Revision: D8886798

Pulled By: malfet

fbshipit-source-id: 52a8b4df85f7e350e36c842ee4428f3a1cba2588
goodlux pushed a commit to goodlux/pytorch that referenced this pull request Aug 15, 2018
…ch#9230)

Summary:
Fix RoIAlignOp GPU implementation for RoIs without batch index
According to https://caffe2.ai/docs/operators-catalogue.html#roialign, RoIs is "2D input of shape (R, 4 or 5)"
Pass RoIs 2nd dimension as kernel parameter and adjust kernel accordingly
Pull Request resolved: pytorch#9230

Reviewed By: houseroad

Differential Revision: D8886798

Pulled By: malfet

fbshipit-source-id: 52a8b4df85f7e350e36c842ee4428f3a1cba2588
@ezyang ezyang added the merged label Jun 26, 2019
@malfet malfet deleted the fix_roi_align_on_gpu branch April 17, 2022 19:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants