pytorch-faster-rcnn icon indicating copy to clipboard operation
pytorch-faster-rcnn copied to clipboard

Published 20 hours ago • verified publisher icon AlphaJia

rel_codes can not reshape bug

Open fatalfeel opened this issue 3 years ago • 0 comments

pred_boxes = self.decode_single(rel_codes.reshape(box_sum, -1), concat_boxes) => rel_codes = rel_codes.view(box_sum, -1) pred_boxes = self.decode_single(rel_codes, concat_boxes)

########################bug ########################################### the rel_codes = rel_codes.view(box_sum, -1) #reshape fail on fpn resnet-50

########################correct###################################### rel_codes = rel_codes[:box_sum] #this will be correct

fatalfeel avatar Apr 25 '21 10:04 fatalfeel