小编Min*_*ặng的帖子

我正在尝试加载 YOLOv5 模型并使用它来预测特定图像。我的问题是我想在我的应用程序中显示带有边界框的预测图像，因此我需要直接从 PyTorch 的预测方法获取它以在我的应用程序中显示。

    model = torch.hub.load('yolov5', 'custom', path=model_name, force_reload=True, 
    source='local')
    pred = model(image)
    pred.show() #show image but can't assign to a variable
    pred.save() #save image to runs\detect\exp

我想要这样的东西：

    predict_image = model(image)
    cv2.imshow('Predict', predict_image)

谢谢。