cv2 error

[didnmj]

joon@DESKTOP-J3EB27K:~/Complex-YOLOv3$ python test_both_side_detection.py --split=sample --folder=sampledata --save_video=true
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=True, 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 51, in detect_and_draw
bev_utils.drawRotatedBox(display_bev, x, y, w, l, yaw, cnf.colors[int(cls_pred)])
File "/home/joon/Complex-YOLOv3/utils/kitti_bev_utils.py", line 174, in drawRotatedBox
cv2.line(img, (corners_int[0, 0], corners_int[0, 1]), (corners_int[3, 0], corners_int[3, 1]), (255, 255, 0), 2)
cv2.error: OpenCV(4.5.5) 👎 error: (-5:Bad argument) in function 'line'

Overload resolution failed:

• Can't parse 'pt1'. Sequence item with index 0 has a wrong type
• Can't parse 'pt1'. Sequence item with index 0 has a wrong type

[didnmj]

kitti_bev_utils.py line 169 ~ 174
//////////////////////////////////////
def drawRotatedBox(img,x,y,w,l,yaw,color):
bev_corners = get_corners(x, y, w, l, yaw)
corners_int = bev_corners.reshape(-1, 1, 2).astype(int)
cv2.polylines(img, [corners_int], True, color, 2)
corners_int = bev_corners.reshape(-1, 2)
cv2.line(img, (corners_int[0, 0], corners_int[0, 1]), (corners_int[3, 0], corners_int[3, 1]), (255, 255, 0), 2)

[Shibaditya99]

following

[sjkaang]

I have the same problem.. Have you solved it?

[sunny553]

I have the same problem.. Have you solved it?

[sjkaang]

I have the same problem.. Have you solved it?

I have solved this. Change as below

kitti_bev_utils.py line 174
///////////////////////////////////
cv2.line(img, (int(corners_int[0, 0]), int(corners_int[0, 1])), (int(corners_int[3, 0]), int(corners_int[3, 1])), (255, 255, 0), 2)

But I'm not sure why 'astype' didn't work in line 171 >>> "corners_int = bev_corners.reshape(-1, 1, 2).astype(int)"