小编LSB*_*LSB的帖子

我目前正在尝试使用我自己的脚本而不是 detector.py 从我的图像中使用自定义模型获取边界框坐标。我想获取在图像上绘制边界框所需的坐标。有人可以帮我吗？

model = torch.hub.load('ultralytics/yolov5', 'custom', 'best.pt')
model = model.autoshape()

results = model(img, size=416)