关于yolov5-v7模型的自定义模型加载测试问题

Posted 2022-08-09 刘润森！

@Author：Runsen

本想找下，有没有代码可以实现自定义模型的检测，发现百度没有一篇。

大家肯定说，直接用load加载不就是行了吗，别以为用torch.load(“”) 直接加载就Ok？ torch.load 加载是你通过torch训练的，不是yolo训练，那肯定不同

可怜的我去GitHub的issue找到了答案。

官方模型加载yolov5预训练模型是

import torch

# Model 下载
model = torch.hub.load(\'ultralytics/yolov5\', \'yolov5s\')  # or yolov5n - yolov5x6, custom

# Images
img = \'https://ultralytics.com/images/zidane.jpg\'  # or file, Path, PIL,