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

@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,
  

 

  
1
  
2
  
3
  
4
  
5
  
6
 

文章来源: maoli.blog.csdn.net，作者：刘润森！，版权归原作者所有，如需转载，请联系作者。

原文链接：maoli.blog.csdn.net/article/details/126168382