0
  • 聊天消息
  • 系统消息
  • 评论与回复
登录后你可以
  • 下载海量资料
  • 学习在线课程
  • 观看技术视频
  • 写文章/发帖/加入社区
会员中心
创作中心

完善资料让更多小伙伴认识你,还能领取20积分哦,立即完善>

3天内不再提示

关于YOLOU中模型的测试

新机器视觉 来源:新机器视觉 作者:新机器视觉 2022-08-16 10:24 次阅读

这里推荐一个YOLO系列的算法实现库YOLOU,此处的“U”意为“United”的意思,主要是为了学习而搭建的YOLO学习库,也借此向前辈们致敬,希望不被骂太惨;

整个算法完全是以YOLOv5的框架进行,主要包括的目标检测算法有:YOLOv3、YOLOv4、YOLOv5、YOLOv5-Lite、YOLOv6、YOLOv7、YOLOX以及YOLOX-Lite。

同时为了方便算法的部署落地,这里所有的模型均可导出ONNX并直接进行TensorRT等推理框架的部署,后续也会持续更新。

01

模型精度对比

554f487a-1cec-11ed-ba43-dac502259ad0.gif   服务端模型 这里主要是对于YOLO系列经典化模型的训练对比,主要是对于YOLOv5、YOLOv6、YOLOv7以及YOLOX的对比,部分模型还在训练之中,后续所有预训练权重均会放出,同时对应的ONNX文件也会给出,方便大家部署应用落地。 注意,这里关于YOLOX也没完全复现官方的结果,后续有时间还会继续调参测试,尽可能追上YOLOX官方的结果。 下表是关于YOLOU中模型的测试,也包括TensorRT的速度测试,硬件是基于3090显卡进行的测试,主要是针对FP32和FP16进行的测试,后续的TensorRT代码也会开源。目前还在整理之中。 5569fd78-1cec-11ed-ba43-dac502259ad0.png轻量化模型 为了大家在手机端或者其他诸如树莓派瑞芯微AID以及全志等芯片的部署,YOLOU也对YOLOv5和YOLOX进行了轻量化设计。 下面主要是对于边缘端使用的模型进行对比,主要是借鉴之前小编参与的YOLOv5-Lite的仓库,这里也对YOLOX-Lite进行了轻量化迁移,总体结果如下表所示,YOLOX-Lite基本上可以超越YOLOv5-Lite的精度和结果。 55806cac-1cec-11ed-ba43-dac502259ad0.png

02

如何使用YOLOU?

554f487a-1cec-11ed-ba43-dac502259ad0.gif   安装 这里由于使用的是YOLOv5的框架进行的搭建,因此安装形式也及其的简单,具体如下:

gitclonehttps://github.com/jizhishutong/YOLOU
cdYOLOU
pipinstall-rrequirements.txt
数据集 这里依旧使用YOLO格式的数据集形式,文件夹形式如下:
train:../coco/images/train2017/
val:../coco/images/val2017/
具体的标注文件和图像list如下所示:
├──images#xx.jpgexample
│├──train2017
││├──000001.jpg
││├──000002.jpg
││└──000003.jpg
│└──val2017
│├──100001.jpg
│├──100002.jpg
│└──100003.jpg
└──labels#xx.txtexample
├──train2017
│├──000001.txt
│├──000002.txt
│└──000003.txt
└──val2017
├──100001.txt
├──100002.txt
└──100003.txt
参数配置 YOLOU为了方便切换不同模型之间的训练,这里仅仅需要配置一个mode即可切换不同的模型之间的检测和训练,具体意义如下: 55a63f90-1cec-11ed-ba43-dac502259ad0.png 注意:这里的mode主要是对于Loss计算的选择,对于YOLOv3、YOLOv4、YOLOv5、YOLOR以及YOLOv5-Lite直接设置mode=yolo即可,对于YOLOX以及YOLOX-Lite则设置mode=yolox,对于YOLOv6和YOLOv7则分别设置mode=yolov6和mode=yolov7; 注意由于YOLOv7使用了Aux分支,因此在设置YOLOv7时有一个额外的参数需要配置,即use_aux=True。 具体训练指令如下:
pythontrain.py--modeyolov6--datacoco.yaml--cfgyolov6.yaml--weightsyolov6.pt--batch-size32
检测指令如下:
pythondetect.py--source0#webcam
file.jpg#image
file.mp4#video
path/#directory
path/*.jpg#glob
'https://youtu.be/NUsoVlDFqZg'#YouTube
'rtsp://example.com/media.mp4'#RTSP,RTMP,HTTPstream

审核编辑:彭静
声明:本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人,不代表电子发烧友网立场。文章及其配图仅供工程师学习之用,如有内容侵权或者其他违规问题,请联系本站处理。 举报投诉
  • 硬件
    +关注

    关注

    11

    文章

    3252

    浏览量

    66110
  • 模型
    +关注

    关注

    1

    文章

    3171

    浏览量

    48711
  • 代码
    +关注

    关注

    30

    文章

    4744

    浏览量

    68343

原文标题:汇集YOLO系列所有算法,YOLOU算法实现库来啦

文章出处:【微信号:vision263com,微信公众号:新机器视觉】欢迎添加关注!文章转载请注明出处。

收藏 人收藏

    评论

    相关推荐

    基于模型的动态测试工具TPT

    针对嵌入式系统的基于模型的动态测试工具,支持众多业内主流的工具平台和测试环境,可应用于整个嵌入式软件开发周期,实现各种异构环境下的自动化测试。无论是在
    发表于 02-20 14:28 916次阅读
    基于<b class='flag-5'>模型</b>的动态<b class='flag-5'>测试</b>工具TPT

    关于multisim仿真模型的建立

    ADA4084和ADA4096一样是双运放,因此在建立仿真模型时,复制了ADA4096的元件符号(分为A和B两个单运放), 之后在进行节点匹配时,在cir文件中看到了如下: 在上面的cir文件可以
    发表于 08-29 10:33

    关于EMSCAN RFX2天线模型测试系统的介绍

    关于EMSCAN RFX2天线模型测试系统的介绍
    发表于 05-20 07:21

    动态模型在软件系统测试过程中的应用研究

    系统测试是软件开发过程的重要环节,系统测试过程的动态模型有助于更好地理解和分析系统行为,做出正确的判断和决策;相对于已有的软件测试
    发表于 07-16 11:58 8次下载

    关于HTSM(启发式策略模型)的详细介绍

    关于HTSM(启发式策略模型)的详细介绍,可以应用于测试设计。
    发表于 05-19 11:03 4次下载

    ESD模型测试标准

    ESD模型测试标准
    发表于 12-10 14:02 18次下载

    关于云计算的实体论模型的分析与评价_孙红

    关于云计算的实体论模型的分析与评价_孙红
    发表于 03-19 11:30 1次下载

    一种新的软件测试模型—软件层次化模型

    针对基于Markov链模型的软件测试技术在测试数据生成时不考虑软件的结构信息,生成的测试数据集对代码路径的覆盖能力以及缺陷检测能力都较低的问题,将统计
    发表于 12-08 11:02 0次下载
    一种新的软件<b class='flag-5'>测试</b><b class='flag-5'>模型</b>—软件层次化<b class='flag-5'>模型</b>

    关于 AD 如何创建 3D 模型及设计教程 Ver1.0

    关于Altium_Design如何创建3D模型及3D设计教程Ver1.0
    发表于 01-25 17:24 29次下载

    关于模型测试与持续集成相结合的可行性分析

    Jenkins 可以通过调用操作系统 shell 命令运行 MATLAB 脚本,关键点是如何将 Simulink 模型测试过程集成到 MATLAB 脚本,好在 MATLAB 提供了单元测试
    的头像 发表于 09-16 08:42 1886次阅读
    <b class='flag-5'>关于</b><b class='flag-5'>模型</b><b class='flag-5'>测试</b>与持续集成相结合的可行性分析

    辐射测试Antenna与EUT的测试距离换算

    关于辐射测试Antenna与EUT的测试距离换算,不同地区的标准或者不同产品经常会遇到关于天线与产品的
    的头像 发表于 08-14 16:07 6431次阅读
    辐射<b class='flag-5'>测试</b><b class='flag-5'>中</b>Antenna与EUT的<b class='flag-5'>测试</b>距离换算

    AN41-关于SPICE宏模型库的问答

    AN41-关于SPICE宏模型库的问答
    发表于 04-17 15:04 3次下载
    AN41-<b class='flag-5'>关于</b>SPICE宏<b class='flag-5'>模型</b>库的问答

    直播回顾 |车辆动力学模型在仿真测试的应用实践

    车辆动力学模型是仿真测试系统的核心组成部分,本次研讨会结合经纬恒润的实际工程经验,介绍车辆动力学模型在典型仿真测试系统的应用及实践,探讨在
    的头像 发表于 03-30 15:48 696次阅读
    直播回顾 |车辆动力学<b class='flag-5'>模型</b>在仿真<b class='flag-5'>测试</b><b class='flag-5'>中</b>的应用实践

    Simulink集成模型测试太慢怎么办?

    作者|单线程生物小编|吃不饱Tips:现阶段模型开发大部分采用Simulink,为了验证模型实现了相关功能,需要对模型进行测试模型
    的头像 发表于 09-09 14:53 1150次阅读
    Simulink集成<b class='flag-5'>模型</b><b class='flag-5'>测试</b>太慢怎么办?

    摩尔线程与智谱AI完成大模型性能测试与适配

    近日,摩尔线程与智谱AI在人工智能领域开展了一轮深入的合作,共同对GPU大模型进行了适配及性能测试。此次测试不仅涵盖了大模型的推理能力,还涉及了基于摩尔线程夸娥(KUAE)千卡智算集群
    的头像 发表于 06-14 16:40 1012次阅读