位置: 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年我要实现的目标是)

  • 苹果电脑没有wifi图标怎么办(苹果电脑没有wifi图标)

    苹果电脑没有wifi图标怎么办(苹果电脑没有wifi图标)

  • 实况照片怎么拍久一点(苹果实况照片怎么拍)

    实况照片怎么拍久一点(苹果实况照片怎么拍)

  • 拼多多商家地址在哪里查看(拼多多商家地址在哪)

    拼多多商家地址在哪里查看(拼多多商家地址在哪)

  • 抖音怎么出合集(抖音怎么出合集啊)

    抖音怎么出合集(抖音怎么出合集啊)

  • p40中国开售时间(p40国内什么时候卖)

    p40中国开售时间(p40国内什么时候卖)

  • 手机摔了屏幕没坏黑屏(手机摔了屏幕没反应怎么办)

    手机摔了屏幕没坏黑屏(手机摔了屏幕没反应怎么办)

  • vip抢票一定能抢到吗(vip抢票一次多少钱)

    vip抢票一定能抢到吗(vip抢票一次多少钱)

  • word标题前的黑点怎么添加(word标题前的黑点会打印出来吗)

    word标题前的黑点怎么添加(word标题前的黑点会打印出来吗)

  • 5g优享服务是什么意思(5g优享网络服务是什么)

    5g优享服务是什么意思(5g优享网络服务是什么)

  • 拼多多推广余额怎么退(拼多多推广余额可以提现吗)

    拼多多推广余额怎么退(拼多多推广余额可以提现吗)

  • 苹果手机没反应也无法关机(苹果手机没反应了也重启不了)

    苹果手机没反应也无法关机(苹果手机没反应了也重启不了)

  • 华为下拉通知栏不见了(华为下拉通知栏没反应)

    华为下拉通知栏不见了(华为下拉通知栏没反应)

  • 骁龙八核处理器相当于什么(骁龙八核处理器排行)

    骁龙八核处理器相当于什么(骁龙八核处理器排行)

  • vivos5支持nfc吗(vivos5支持无线充吗)

    vivos5支持nfc吗(vivos5支持无线充吗)

  • 美团点评退款是什么意思(美团点评退款是怎么回事)

    美团点评退款是什么意思(美团点评退款是怎么回事)

  • 5v200ma能用5v1a充吗(5v200ma充电器有哪些)

    5v200ma能用5v1a充吗(5v200ma充电器有哪些)

  • 小米wifi5g是什么意思(小米wifi的5g和4g的区别)

    小米wifi5g是什么意思(小米wifi的5g和4g的区别)

  • 快手唱歌需要什么设备(快手唱歌需要什么软件)

    快手唱歌需要什么设备(快手唱歌需要什么软件)

  • 显卡怎么检查是否坏了(显卡怎么检查是否损坏)

    显卡怎么检查是否坏了(显卡怎么检查是否损坏)

  • 小米20000充电宝怎么拆(小米20000充电宝真假辨别)

    小米20000充电宝怎么拆(小米20000充电宝真假辨别)

  • 安卓软件开发什么语言(开发安卓软件需要学什么)

    安卓软件开发什么语言(开发安卓软件需要学什么)

  • 博朗温度计怎么调整摄氏度(博朗温度计怎么调华氏温度和摄氏温度)

    博朗温度计怎么调整摄氏度(博朗温度计怎么调华氏温度和摄氏温度)

  • 抖音激萌特效在哪里(抖音激萌特效在哪里找)

    抖音激萌特效在哪里(抖音激萌特效在哪里找)

  • 美团众包保证金退到哪里(美团众包保证金可以不交吗)

    美团众包保证金退到哪里(美团众包保证金可以不交吗)

  • 长期零申报的公司容易注销吗
  • 总账会计和全盘会计的区别
  • 以货换货要开发票吗
  • 预缴增值税的附加税
  • 会计核算原则主要包括
  • 存货项目包括哪些具体内容
  • 合伙做生意账目
  • 金融企业计提的坏账准备金
  • 居民企业只就其境内全部所得纳税
  • 随同产品销售单独计价的包装物计入
  • 以前年度固定资产入账错误怎么办?
  • 车辆保险费的车船税计入什么会计科目
  • 借款合同 增值税
  • 资产类账户的期末余额计算公式
  • 外籍员工个税信息采集
  • 同一控制下的企业合并的会计核算方法
  • 包装物押金如何确定销售额?
  • 未分配利润是什么类科目
  • 增值税普通发票有什么用
  • 电动车固定资产报废年限是多少年
  • 分公司企业所得税怎么缴纳
  • 丧失控股权的后果
  • 增值税发票跨年了还能用吗
  • 退款给客户怎么写分录
  • 留底的进项税额记哪儿?
  • 其他货币资金属于资产类科目
  • 违约金开什么票据
  • php基础入门教程
  • php自定义函数的关键字是什么
  • 基础知识讲解
  • 我国中小企业成本管理现状
  • 政府购买企业服务是否交税
  • 电子承兑汇票支付信用查询是什么意思
  • softag
  • 暂估增值税可以抵扣吗
  • 教你一行代码
  • php 文件管理系统
  • js生成条形码的方式有哪些
  • 收到投资款的会计科目怎么入账
  • mysql集群三种方式
  • 过来人告诉你:女人最珍贵的三种东西,舍得给你说明爱你
  • js框架jquery
  • 记载资金的账簿
  • 房地产销售折扣
  • phpcms默认密码
  • 邀请客户参加会议
  • mongodb replica set 添加删除节点的2种方法
  • 会计科目是怎么来的
  • access 模糊匹配
  • 小规模纳税人需要每月清卡吗
  • 一般纳税人租赁费税率5%是简易征收吗?
  • 其他综合收益转到留存收益
  • 接受捐赠的固定资产进项税额可以抵扣吗
  • 企业产品研发
  • 外贸企业出口货物的计税依据
  • 预收账款怎么处理
  • 营业外支出科目核算的内容
  • 车辆商业险和交强险不在同一日期
  • 异地出租房屋
  • 员工工资怎么核算
  • 贷款收取服务费
  • 劳务公司开出发票没有收到钱,怎么做成本
  • mac如何隐藏桌面
  • ktpcntr.exe
  • 苹果电脑快捷键截图怎么截
  • linux如何调出鼠标
  • win10qq在哪里启动
  • opengl 三角形
  • webpack使用ejs
  • android.json
  • javascript prototype inheritence
  • js在对象的方法中访问属性name的语法为
  • 分页jsp
  • jquery对象可以直接使用dom吗
  • 安卓app压力测试
  • 国税总局北京市税务局
  • 股权转让怎么计算个人所得税
  • 国税纳税服务有哪些项目
  • 车辆购置税税率多少
  • 食品烟酒消费包括哪些
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

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

    友情链接: 武汉网站建设