YOLOv5 之 torch.hub.load（）检测无结果

前言

在使用torch.hub.load（）加载yolov5模型，进行推理的过程中，我发现，推理结果和在yolo代码中推理结果并不相同。

错误推理代码：

model=torch.hub.load('./yolov5', 'custom', path="./best.pt", source='local',force_reload=True)

outs=model(image, size=640)

正确推理代码：

model=torch.hub.load('./yolov5', 'custom', path="./best.pt", source='local',force_reload=True)

outs=model(image_[:,:,::-1], size=640)

分析：

查遍yolo代码才发现，yolo加载图片时，loadimage函数会进行一个BGR到RGB的转换。但是hub加载就没了呀，所以必须手动转换才可以。