1、介绍
torch.hub 的检测可以实现几行代码进行检测,这里介绍yolov5中利用hub进行推理
import torch
'''
解决下面的问题
Exception: cannot instantiate 'PosixPath' on your system. Cache may be out of date,
try `force_reload=True` or see https://docs.ultralytics.com/yolov5/tutorials/pytorch_hub_model_loading for help.
'''
import pathlib
temp = pathlib.PosixPath
pathlib.PosixPath = pathlib.WindowsPath
model = torch.hub.load('./','custom',path='runs/train/exp/weights/best.pt',source='local')
img = './datasets/images/train/180.jpg'
result = model(img)
result.show()
2、Torchhub 模型加载
具体要使用的如下:
因为yolov5下有hubconf文件,所以我们可以正常使用torchhub