ghimiredhikura / Complex-YOLOv3

PyTorch implementation of Complex-YOLO paper with YoloV3
GNU General Public License v3.0
190 stars 68 forks source link

Fix for RuntimeError: Expected object of scalar type Byte but got scalar type Bool #28

Closed saqibmobin closed 4 years ago

saqibmobin commented 4 years ago

Error: Traceback (most recent call last): File "test_detection.py", line 124, in detections = utils.non_max_suppression_rotated_bbox(detections, opt.conf_thres, opt.nms_thres) File "/aimldl-cod/external/Complex-YOLOv3/utils/utils.py", line 301, in non_max_suppression_rotated_bbox invalid = large_overlap & label_match RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument #2 'other' in call to _th_and

Changes removed uint8 as large_overlap datatype Remarks Fixed the error

ghimiredhikura commented 4 years ago

Thanks.