cascade-rcnn-fpn-faster_rcnn-pytorch1.0 icon indicating copy to clipboard operation
cascade-rcnn-fpn-faster_rcnn-pytorch1.0 copied to clipboard

Published 20 hours ago • verified publisher icon diamour

run shape dismatch

Open gentlebreeze1 opened this issue 3 years ago • 0 comments

Size does not match at dimension 0 expected index [128, 1, 4] to be smaller than src [127, 4, 4] apart from dimension 1. from bbox_pred_select = torch.gather(bbox_pred_view, 1,rois_label.long().view(rois_label.size(0), 1, 1).expand(rois_label.size(0),1, 4)) in fpn.py? how to deal with this bug?@diamour

gentlebreeze1 avatar May 18 '21 08:05 gentlebreeze1