当前位置:网站首页>新工程加载YOLOV6的预训练权重问题
新工程加载YOLOV6的预训练权重问题
2022-08-02 03:21:00 【& Tom】
1. 问题背景
美团发布的YOLOV6又提供了几个Trick,本着学习的态度对YOLOv6进行学习,可是在加载预备训练权重的时候,一直报错,经过一下午+一晚上的调试找到一个简单易实现的方法。
报错问题:
File "/home/anaconda3/lib/python3.7/site-packages/torch/serialization.py", line 426, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "/home/anaconda3/lib/python3.7/site-packages/torch/serialization.py", line 613, in _load
result = unpickler.load()
ModuleNotFoundError: No module named 'yolov6'
2.本着遇见问题先百度的态度,经过一番百度,有以下几个方法:
https://blog.csdn.net/hpccph15/article/details/124181484
ModuleNotFoundError: No module named 'models' · Issue #18325 · pytorch/pytorch · GitHub
我照着做了,但是仍然报错,直到后来,有人在yolov6的github中提出了这个问题,链接:
一旦脱离了yolov6的repo,torch.load('path/to/yolov6s.pt')将失效报错 · Issue #278 · meituan/YOLOv6 · GitHub
在这个issue中,大佬指出了问题的关键,

其实yolov5 6.0版本中,仍然是这么个写法 ,以前用的时候居然每遇到这个问题。
3.知道原因了,那就解决问题
说实话,我不会解决这个问题,后来我一想,既然保存了原来工程文件的配置,那么直接用yolov6的源码进行加载不就行了麻,我采取最苯的办法,将YOLOV6源码中的yolov6文件夹直接拷贝到了我自己的工程中,果然它能正确加载了,就像这样,放到根目录下就可以了。

如果有大佬会解决这个问题,请您教我一下,感激不尽!
边栏推荐
- Redis笔记进阶篇:万字长文-整理Redis,各种知识点,建议收藏
- np.isnan ()
- MySQL分页查询的5种方法
- sh: 1: curl: not found
- RHCSA第三天
- Day34 LeetCode
- MySQL删除表数据 MySQL清空表命令 3种方法
- Daily practice------There are n integers, so that the previous numbers are moved back m positions in order, and the last m numbers become the first m numbers
- np.isnan()
- [详解C语言]一文带你玩转C语言小游戏---三子棋
猜你喜欢
随机推荐
[Basic Tutorial of Remote Control Development 1] Crazy Shell Open Source Formation Drone-GPIO (Remote Control Indicator Light Control)
3分钟带你了解微信小程序开发
知识问答与知识会话的区别
[详解C语言]一文带你玩转C语言小游戏---三子棋
Redis的集群模式
Monaco Editor 的基本用法
HCIP-第十一天-MPLS+BGP
MySQL中JOIN的用法
API 低代码开发:接口大师,一套开发、管理和提供接口的产品框架
ImportError: libGL.so.1: cannot open shared object file: No such file or directory
APK的安装过程分析 PMS包管理器
DSPE-PEG-DBCO 磷脂-聚乙二醇-二苯并环辛炔 一种线性杂双官能聚乙二醇化试剂
JJWT工具类
DAY-1 | 求两个正整数的最大公约数与最小公倍数之和——辗转相除法
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
MySQL中字符串比较大小(日期字符串比较问题)
排序学习笔记(二)堆排序
错误:with open(txt_path,‘r‘) as f: FileNotFoundError: [Errno 2] No such file or directory:
Cloud server installation and deployment of Nacos 2.0.4 version
@Autowired与@Resource区别









