请问YOLO已经训练好了自己的模型文件,.pt的文件,如何写代码调用这个文件,输出检测的数量和准确度等信息。
以下代码不对吗?错误在哪里 现在就是同意的用这个 模型文件 在测试的 yolov5中的 detect.py中运行的结果很好,但是我自己写的下面的代码就不太行,显示出来结果差~
#import torch
#import numpy as np
#import cv2
#model = torch.hub.load('ultralytics/yolov5', 'custom', path='C:\\Users\\Administrator\\Desktop\\ceshi1\\best.pt')
#frame = cv2.imread('C:\\Users\\Administrator\\Desktop\\11\\0.jpg')
#frame = cv2.resize(frame,(640,640))
#results = model(frame)
#cv2.imshow('yolo', np.squeeze(results.render()))
#print(results)
#cv2.waitKey(0)
#
```cv2.destroyAllWindows()