Hi all,
I have just started tinkering with the PyTorch framework.
I am getting an error while using utils library to draw the bounding boxes.
Here is the code:
#draw the bounding boxes around the objects detected in the image
try:
    input_value = img_tensor.cpu() #copying image transformed tensor data from GPU
    plt.imshow(utils.make_grid(input_value, nrow=1).permute(1,2,0)) #display the input image
    plt.show()
    boxes_tensor = torch.Tensor(pred_boxes) #converting boxes into a tensor
    preprocess_img = transforms.Compose([transforms.ToTensor()])(cpy_img) #transforming image to a tensor from the same input file
    out_img_tensor = preprocess_img.squeeze(0)
    out_img_type = out_img_tensor.type(torch.uint8) #converting the flaot tensor into uint8 as required by utils
    final_output = utils.draw_bounding_boxes(out_img_type, boxes_tensor, pred_classes) #draw the bounding boxes
    plt.imshow(utils.make_grid(final_output, nrow=1).permute(1,2,0)) #display the ouput result
    plt.show()
except Exception as er4:
    print(er4)

Please find the below images for reference:

Thank you