Python调用yolo模型

Python调用yolo模型

本文在pycharm实验没问题，在juypter notebook会报错。

导入依赖

#导入依赖
import torch

载入本地模型

C:/Users/Qingchen/yolov5为克隆的yolov5的路径，C:/Users/Qingchen/yolov5/best_0523.pt是自己训练模型的weights的路径，可以替换成官方给的weights测试

#载入本地模型
model = torch.hub.load('C:/Users/Qingchen/yolov5', 'custom', 'C:/Users/Qingchen/yolov5/best_0523.pt', source='local')

载入图片尽量用\\否则容易报错

#载入图片并显示
img = 'C:\\Users\\Qingchen\\yolov5\\data\\Doraemon\\a.jpg'
result = model(img)
result.show()

实验结果

完整代码

import torch
model = torch.hub.load('C:/Users/Qingchen/yolov5', 'custom', 'C:/Users/Qingchen/yolov5/best_0523.pt', source='local')
img = 'C:\\Users\\Qingchen\\yolov5\\data\\Doraemon\\a.jpg'
result = model(img)
result.show()