首页 > 其他分享 >关于yolov5官方项目训练的模型加载时报no module named models..Model/common

关于yolov5官方项目训练的模型加载时报no module named models..Model/common

时间:2022-09-05 09:11:22浏览次数:74  
标签:yolov5 no models 模型 保存 官方 model Model 加载

出现该报错的原因是:训练保存的模型里包含了网络结构,结构的model类直接包含了路径地址model.yolo.Model

解决办法就是重新在官方项目代码中加载模型,并再次只保存其中的模型权重,注意不要保存模型结构。

保存方法如下:

trans = torch.load('你的模型路径')

troch.save(trans['model'].state_dict(), 新的模型名)

 

感兴趣的朋友也可以打印看看官方模型保存的字典结构,每个打印看看

标签:yolov5,no,models,模型,保存,官方,model,Model,加载
From: https://www.cnblogs.com/chentiao/p/16656891.html

相关文章

  • yum安装nginx报错解决nginx: [emerg] unknown directive “stream“ in /etc/nginx/ng
    yum安装nginx,其中看nginx有stream参数,但是nginx-t是报错“nginx:[emerg]unknowndirective“stream“in/etc/nginx/nginx.conf问题”  解决办法#安装nginx源......
  • NOIP复习(一)最短路
    dijkstra复习一遍模板。Dijkstra模板适用性:适用于非负权图。每个点第一次从堆中被取出时,其\(dis\)一定是最短路。\(Dijkstra\)贪心的正确性:现证明:取出\(x\)时,它......
  • NOIP复习(二)二分算法
    提供一种二分写法,不太用考虑边界的问题。intl=st,r=ed,ans=ed+1;while(l<=r){intmid=(l+r)>>1;if(check(mid))ans=mid,l=mid+......
  • 使用 Node.js 构建基于 Typescript 的命令行界面 (CLI)
    使用Node.js构建基于Typescript的命令行界面(CLI)我们的目标是使用Typescript创建一个基于Node.js的命令行界面(CLI)。此设置包括顶级等待支持和ES模块导入。......
  • python中的None
    一:NoneNone是python中的一个特殊的常量,表示一个空的对象。数据为空并不代表是空对象,例如空列表:[],等都不是None。None有自己的数据类型NontType。二:FalsePython中数......
  • gcc-arm-none-eabi工具链工具使用方法及介绍
    为了方便自己在学习过程中查看一些工具的使用方法,将其搬运到自己的创作空间原文链接:STM32高级开发(5)-gcc-arm-none-eabi_Foresights的博客-CSDN博客_arm-none-eabi-gccg......
  • go编译c报错 cc1.exe: sorry, unimplemented: 64-bit mode not compiled in
    go编译c报错cc1.exe:sorry,unimplemented:64-bitmodenotcompiledin说明当前gcc是32位,无法在当前64位机器上正常工作,需要更新gcchttps://sourceforge.net/project......
  • 注解Annotation
    注解是一种引用数据类型,重点掌握Deprecated(表示已过时),Override(表示重写)。元注解是用来标注注解类型的注解如Target(用来标注注解可以出现在哪些位置)、Retention(用来标注最......
  • git ignore忽略规则
    aliases:[gitignore]tags:[git,blog]link:date:2022-09-04目录aliases:[gitignore]tags:[git,blog]Git忽略文件提交的方法Git忽略规则Git忽略规则优先级Gi......
  • 生成allure报告报错:Exception in thread "main" java.lang.NoClassDefFoundError: com
     报错如图:  问题解决经过:从报错中看刚开始以为是java版本问题,然后就尝试换了jdk11(我本地之前是jdk8,也就是jdk1.8),但是换了之后还是不行,然后以为是allure没有装好......