Skip to content

Improve error-checking in RoIPool / Align #1957

@fmassa

Description

@fmassa

Originally reported in #1953

We should raise an error if the box tensor that is fed to roi_align / roi_pool is Nx4. It should be Nx5 after conversion. This will make it easier for users to spot misuse.

cc @zhangguanheng66

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions