前言
在使用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加载就没了呀,所以必须手动转换才可以。