yolov5 多版本共存

在做工业项目时，yolov5是比较好的检测算法，

但是如果一个项目中需要多个版本的yolov5检测项目，检测不同的物体，

加载不同版本的网络会报错

self.weights='yolov5s.pth'
model = torch.load(self.weights)
stride = int(model.stride.max())  # model stride

原因有：

1.网络结构不一样，比如激活函数不一样，层内容不一样

2.检测层Detect不一样。

解决方法：jit.trace方式。

导出：

        self.model.model[-1].export = Trueself.model.eval()self.model=self.model.float()x = torch.randn(1, 3, 544, 960).float().cuda()traced_cell = torch.jit.trace(self.model, (x))torch.jit.save(traced_cell, "jit_face_detect.pth")

加载部分代码：

      face_model_path=r'D:\project\detect\face\yolov5-face-master\jit_face_detect.pth'face_para

yolov5 多版本共存相关推荐

适合win7的python版本_windows下多个python版本共存，如何在Windows7系统上安装最新的64位Python3.6.2...
windows下多个python版本共存,如何在Windows7系统上安装最新的64位Python3.6.2 1.官网下载python3.6.2 https://www.python.org/ftp/ ...
python2和python3分别是python的两个版本_Windows下Python2与Python3两个版本共存的方法详解...
前言一向用Python 3,最近研究微信公众号开发,各云平台只支持Python 2.7,想用其他版本需要自己搭建环境.而网上又搜不到Python 3开发微信公众号的资料.暂打算先使用Python 2 ...
Python多版本共存之pyenv
2019独角兽企业重金招聘Python工程师标准>>> 经常遇到这样的情况: 系统自带的Python是2.6,自己需要Python 2.7中的某些特性: 系统自带的Python是2. ...
Linux下快速静态编译Qt以及Qt动态/静态版本共存
Qt下静态编译Qt,根据我的经验,如果按照Windows下那种直接拿官方sdk安装之后的文件来编译是行不通的,需要直接下载Qt的source包,目前诺基亚的源码叫做qt-everywhere-open ...
APP多版本共存，服务端如何兼容？
做过APP产品的技术人员都知道,APP应用属于一种C/S架构的,所以在做多版本兼容,升级等处理则比较麻烦,不像web应用那么容易.下面将带大家分析几种常见的情况和应对方式: 小改动或者新加功能的 ...
CentOS下与Apache连接的PHP多版本共存方案实现详解
这篇文章主要介绍了CentOS下与Apache连接的PHP多版本共存方案实现,针对mod_fcgi模块的配置作了一些说明,需要的朋友可以参考下在apache下整合fastCGI模式运行的php-fp ...
Android配置build.gradle解锁更高逼格玩法（多版本共存、分服务器打包等）
背景在日常的AndroidAPP开发过程中,常常需要在同一设备上安装同一APP的多个版本.如在没有测试设备的情况下需要使用到个人设备进行测试,而该设备已安装正式版本且有大量缓存数据.applicat ...
ubuntu php多版本共存切换
做开发时,由于本机开发的php版本跟线上发布的php版本不一致,很容易在上线后,发现因版本的影响导致一些bug,但又不想重新去换本机的php版本,那么多版本共存就很方便了!有必要时,切换到指定版本测试 ...
Centos+Nginx+PHP7.0编译安装(和PHP5.6老版本共存)
来源:http://blog.csdn.net/liuxinmingcode/article/details/50319145 场景 LNMP 当前版本:PHP 5.5.7 为了体验PHP7的速度 ...

yolov5 多版本共存

yolov5 多版本共存相关推荐

最新文章

热门文章