Help！！！cv2.error: OpenCV(4.5.2) :-1: error: (-5:Bad argument) in function 'rectangle' about pytorch-yolo-v3

[VasLem]

At that point c1 and c2 are still tensors, so OpenCV raises type error. To fix it change lines 286 and 287 to:
c1 = tuple(x[1:3].cpu().int().numpy())
c2 = tuple(x[3:5].cpu().int().numpy())

[ray7jq]

c1 and c2 are tensors,you could use like c1[0].item() to fix it.
cv2.rectangle(img, (c1[0].item(),c1[1].item()), (c2[0].item(),c2[1].item()), color, 2)

[syed-javed]

Hi Zxl1556180489,
Did you solve this issue? I am getting the same issue.

[expnn]

x = x.detach().cpu().int().numpy()
img = results[x[0]]
c1 = (x[1], x[2])
c2 = (x[3], x[4])
cls = int(x[-1])

[Roboskeletron]

Replace lines 178 and 179 with this

c1 = tuple(map(torch.Tensor.item, x[1:3].int()))
c2 = tuple(map(torch.Tensor.item, x[3:5].int()))