Skip to content

Commit e091a07

Browse files
sstsai-adlfacebook-github-bot
authored andcommitted
fix rotated bounding box tracing inference error
Summary: Pull Request resolved: #4249 Fixes run time error: RuntimeError: Output 0 of SelectBackward0 is a view and is being modified inplace. This view is the output of a function that returns multiple views. Such functions do not allow the output views to be modified inplace. You should replace the inplace operation by an out-of-place one. Reviewed By: zhanghang1989 Differential Revision: D36433376 fbshipit-source-id: e337e63e2ce09632d42db79a7b6e7b050ceb325c
1 parent 211efad commit e091a07

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

detectron2/modeling/roi_heads/rotated_fast_rcnn.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ def fast_rcnn_inference_rotated(
8080
return [x[0] for x in result_per_image], [x[1] for x in result_per_image]
8181

8282

83+
@torch.no_grad()
8384
def fast_rcnn_inference_single_image_rotated(
8485
boxes, scores, image_shape, score_thresh, nms_thresh, topk_per_image
8586
):

0 commit comments

Comments
 (0)