A possible bug in https://github.com/pytorch/pytorch/blob/master/caffe2/operators/roi_pool_op.cc

vision
1

torch_a_bug
torch_a_bug1076×733 57 KB

As the screenshort, the position of the red frame should be -INFINITY. Secondly, if the output1 is -INFINITY, the output2, argmax, should be the first index in the rois, not be -1.

2

Could you create an issue on GitHub so that the code owners could have a look at it and discuss the issue, please?