I’m trying to draw a bounding box over an image using the draw_bounding_boxes function but am faced with this error.
Here is the code:
img = cv2.imread(img_path, 0)
img = torch.unsqueeze(torch.Tensor(img), 0)
img = img.type(torch.uint8)
res = draw_bounding_boxes(img, bbox)

Can anyone help with this?