search

ghimiredhikura / Complex-YOLOv3

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

I got runtime error issue #14

Open dotorl127 opened 4 years ago

dotorl127 commented 4 years ago

I've clone repo and just typed test command witch $ python test_both_side_detection.py --split=sample --folder=sampledata (is it right command?) and I got runtime error below

moon@moon-desktop:~/Complex-YOLOv3$ python test_both_side_detection.py --split=sample --folder=sampledata Namespace(class_path='data/classes.names', conf_thres=0.8, folder='sampledata', img_size=608, model_def='config/complex_yolov3.cfg', nms_thres=0.4, save_video=False, split='sample', weights_path='checkpoints/yolov3_ckpt_epoch-298.pth') Load TESTING samples from data/KITTI/object/sampledata Done: total TESTING samples 12 Traceback (most recent call last): File "test_both_side_detection.py", line 90, in front_bev_result, img_detections = detect_and_draw(model, front_bevs, Tensor, True) File "test_both_side_detection.py", line 28, in detect_and_draw detections = utils.non_max_suppression_rotated_bbox(detections, opt.conf_thres, opt.nms_thres) File "/home/moon/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

How can I solve this problem? Best regards.

ghimiredhikura commented 4 years ago

Hi @dotorl127,

I guess this is because of pytorch version. It is recommended to use v1.1.0, otherwise you have to fix the syntax error according to version. Refer to #3 for solving your issue.

Best Deepak