位置: IT常识 - 正文

yolo-pose环境搭建及训练和测试(yolov4环境搭建)

编辑:rootadmin
yolo-pose环境搭建及训练和测试 文章目录前言1、准备工作2、训练模型2.1.可能报的错:AttributeError: Cant get attribute SPPF on module models.common2.2.训练模型2.3.测试模型可视化总结前言

推荐整理分享yolo-pose环境搭建及训练和测试(yolov4环境搭建),希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:yolox部署,yolo使用场景,yolo环境配置,yolov3环境搭建,yolov5环境搭建,yolov4环境搭建,yolov3环境搭建,yolov5环境搭建,内容如对您有帮助,希望把文章链接给更多的朋友!

 扔掉学术偏见,拥抱工程化的Yolo。由于一直没有使用过yolo,因此本文旨在从0基础配置yolo环境并训练和测试。  论文地址  代码地址

1、准备工作首先下载coco2017数据集,怎么下载这个我就不详细介绍了,但注意的是,虽然原始coco数据集的文件夹名称也为coco,但为了和yolo所使用的进行区分,请将coco数据集命名为coco2017。然后下载原始coco格式的标签并解压,解压后请务必将person_keypoints_val2017.json文件拷贝到coco/annotations/下,否则后续评估报错!新建一个coco_kpts文件夹,下载yolo格式的关键点检测的标签并解压到coco_kpts文件夹内(谷歌云盘,没有梯子的自己想想办法把,别私聊问我要了);同时创建软连接或者将coco2017中的images和annotations两个文件夹拷贝到coco_kpts下面。最终coco_kpts/如下图:先忽略train2017.cache,后续在训练时会生成。

 最后下载代码并解压至和coco_kpts同级目录下即可。

 在准备完成后,最终的一个文件树为:

yolo-pose环境搭建及训练和测试(yolov4环境搭建)

 至此,便可以训练yolo-pose。

2、训练模型2.1.可能报的错:AttributeError: Cant get attribute SPPF on module models.common

 参考链接  在model/common.py文件中加上如下代码即可。

import warningsclass SPPF(nn.Module): # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher def __init__(self, c1, c2, k=5): # equivalent to SPP(k=(5, 9, 13)) super().__init__() c_ = c1 // 2 # hidden channels self.cv1 = Conv(c1, c_, 1, 1) self.cv2 = Conv(c_ * 4, c2, 1, 1) self.m = nn.MaxPool2d(kernel_size=k, stride=1, padding=k // 2) def forward(self, x): x = self.cv1(x) with warnings.catch_warnings(): warnings.simplefilter('ignore') # suppress torch 1.9.0 max_pool2d() warning y1 = self.m(x) y2 = self.m(y1) return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))2.2.训练模型python train.py --data coco_kpts.yaml --cfg yolov5s6_kpts.yaml --batch-size 64 --img 640 --kpt-label

 需要训练300个epoch,等训练完我会补充下结果,目前来看,效果果然好,3个epoch情况下达到了oks达到了38.3,恐怖。:   训练了100epoch,oks在测试集精度为46。官方汇报的是57,训了300epoch。

2.3.测试模型

 首先在readme中下载一个训练好的权重,如下图,点击那个Yolo5s6_pose_640,将下载好的.pt放到edgeai-yolov5-yolo-pose文件夹内。

 测试命令:

python test.py --data coco_kpts.yaml --img 640 --conf 0.001 --iou 0.65 --weights "/path/to/edgeai-yolov5-yolo-pose/last.pt" --kpt-label

 测试结果:59.9比官方汇报的高两个点。

可视化

 测试/训练的可视化图都在runs文件夹内,我这里放几张测试图例:

总结

 有问题欢迎+vx: wulele2541612007,后续有空会出yolo-pose的源码解读,敬请期待。

本文链接地址:https://www.jiuchutong.com/zhishi/300746.html 转载请保留说明!

上一篇:业务架构图(人工智能业务架构图)

下一篇:2023年底,我要通过这5点,实现博客访问量500W(2023年我要实现的目标是)

  • iphone13pro屏幕是2k的吗(13pro屏幕尺寸多大)

    iphone13pro屏幕是2k的吗(13pro屏幕尺寸多大)

  • 北京健康宝怎么开美颜(北京健康宝怎么申请绿码)

    北京健康宝怎么开美颜(北京健康宝怎么申请绿码)

  • 天猫会员店和天猫超市有什么区别(天猫会员店和天猫88 vip一样吗)

    天猫会员店和天猫超市有什么区别(天猫会员店和天猫88 vip一样吗)

  • 小米mix3消息通知在哪里

    小米mix3消息通知在哪里

  • 淘宝定制商品退货规定(淘宝定制商品退货小二判买家输了)

    淘宝定制商品退货规定(淘宝定制商品退货小二判买家输了)

  • 微博特别关心对方会知道吗(微博里特别关心有消息怎么通知)

    微博特别关心对方会知道吗(微博里特别关心有消息怎么通知)

  • 苹果11有闪退现象是怎么回事(苹果11闪退app)

    苹果11有闪退现象是怎么回事(苹果11闪退app)

  • 小米手机短信闪退解决方法(小米手机短信闪退怎么回事)

    小米手机短信闪退解决方法(小米手机短信闪退怎么回事)

  • 爱剪辑可以合并视频吗(爱剪辑可以合并音频吗)

    爱剪辑可以合并视频吗(爱剪辑可以合并音频吗)

  • 手机卡换了手机后就查不到联系人(手机卡换了手机微信怎么登录)

    手机卡换了手机后就查不到联系人(手机卡换了手机微信怎么登录)

  • 华为手表要一直连接蓝牙吗(华为手表要一直充电吗)

    华为手表要一直连接蓝牙吗(华为手表要一直充电吗)

  • 打印机需要网络接口吗(打印机需要网络密码)

    打印机需要网络接口吗(打印机需要网络密码)

  • 华为p20lite有nfc功能吗(华为p20nfc功能)

    华为p20lite有nfc功能吗(华为p20nfc功能)

  • access2010属于什么数据库(access2010是()中的一个组件,它是主要作用是())

    access2010属于什么数据库(access2010是()中的一个组件,它是主要作用是())

  • 手机文件管理里的文档能删除吗(手机文件管理里怎么新建文件夹)

    手机文件管理里的文档能删除吗(手机文件管理里怎么新建文件夹)

  • 电信3g如何升级4g(电信3g如何升级4g套餐)

    电信3g如何升级4g(电信3g如何升级4g套餐)

  • 无线话筒被锁怎么解锁(无线话筒锁定怎么解锁)

    无线话筒被锁怎么解锁(无线话筒锁定怎么解锁)

  • iphone8双卡双待吗(苹果八双卡双待)

    iphone8双卡双待吗(苹果八双卡双待)

  • 苹果8死机了怎么重启(iphone 8死机)

    苹果8死机了怎么重启(iphone 8死机)

  • 钉钉已注册怎么激活(钉钉已注册怎么注销账号)

    钉钉已注册怎么激活(钉钉已注册怎么注销账号)

  • 淘宝超过七天怎么退货(淘宝超过七天怎样关闭退换货端口)

    淘宝超过七天怎么退货(淘宝超过七天怎样关闭退换货端口)

  • 拼多多怎么查店铺地址(拼多多怎么查店家是哪里的)

    拼多多怎么查店铺地址(拼多多怎么查店家是哪里的)

  • 用了嗨来电怎么恢复正常(使用嗨来电怎么设置)

    用了嗨来电怎么恢复正常(使用嗨来电怎么设置)

  • 快手24小时关注人数上限怎么解除(快手24小时关注人数上限是多少人)

    快手24小时关注人数上限怎么解除(快手24小时关注人数上限是多少人)

  • 如何开始编程语言?给初学者的几个小贴士(怎么开始编程)

    如何开始编程语言?给初学者的几个小贴士(怎么开始编程)

  • 深入理解TDNN(Time Delay Neural Network)——兼谈x-vector网络结构(深入理解linux内核)

    深入理解TDNN(Time Delay Neural Network)——兼谈x-vector网络结构(深入理解linux内核)

  • 增值税留抵退税是什么意思
  • 小规模纳税人的企业所得税税率
  • 预付开发票加油后还能开吗
  • 期初未缴税额本年累计是什么意思
  • 会计凭证的内容中应包括会计分录
  • 税务登记后怎么领发票
  • 应收账款减值处理
  • 房地产企业工程款支付流程
  • 纳税调整的情况
  • 二手房增值税怎么收取
  • 劳务派遣差额征税政策
  • 企业所得税清算报备是什么意思
  • 没有成本票怎么做成本
  • 未实缴出资的股权转让账务处理
  • 维修开票单位写什么
  • 预借现金和现金分期的区别
  • 企业收到政府拆迁补偿款要交税吗怎么做账
  • 如何进行网络测试网速测试
  • 车辆哪些费用计税
  • 在window系统中哪些用户可以查看日志
  • 购货方收到红字发票计入进项税转出还是进项税额负数?
  • 押金要不回来怎么办
  • 巴黎先贤祠的名言
  • 城建税征税范围广吗
  • 长期待摊费用是非流动资产吗
  • 营业外支出会计准则
  • 微信小程序做ppt
  • 销售费用期末余额
  • 微信登录接口购买
  • crontab命令用法
  • chage -l命令
  • 美国人用什么英语词典
  • 关于幼儿园国培的要求和建议
  • 预缴增值税需要预缴城建税吗
  • 筹办期间发生的广告费和业务宣传费可以扣除吗
  • 租店铺需要什么流程需要注意什么事项?
  • 劳务费合同模版
  • 资产负债表的负债项目显示了企业所负担债务的
  • 净利润除以实收资本是什么
  • 未开票收入是怎么算的
  • 浅谈一下新冠的好处
  • sqlserver2005安装后怎么打开
  • 基本户和一般户可以互相转账吗
  • MYSQL数据库使用教程
  • 物业管理企业会计核算的特点如何?
  • 收付实现制的适用范围
  • 企业筹建期间的费用处理
  • 库存商品入库的会计分录怎么写
  • 应收票据及应收账款包括哪些科目
  • 附加税怎么算出来的
  • 差旅费中火车票的进项税怎么计算
  • 房地产企业的会计核算
  • 外经证的办理流程
  • 什么叫小微企业主?
  • 向个人借款在现场怎么写
  • 记帐凭证会计处理程序及其应用
  • 税务登记证办理流程
  • 记账凭证负数怎么填写样板
  • 在sql server中使用对象资源管理器和SQL
  • solaris安装教程
  • win8.1 升级
  • centos nis
  • nginx sbin目录
  • Apache 2.0.55 for Linux 下载
  • linux 翻译
  • [置顶]bilinovel
  • unity错误cs1002
  • 深入理解中国式现代化论文
  • 获取nodejs命令行信息
  • 深入理解javascript特性.pdf
  • unity collider类属性方法
  • 深入浅出nginx实战
  • android(8) ViewPager页面滑动切换
  • js实现@功能
  • 地税局局长权利大吗
  • 国家河南税务局
  • 青岛市市南区税务局第一税务所
  • 成品油消费税税目
  • 爱信诺开票系统怎么安装
  • 发票红冲产生费用吗?
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设