首页IT科技yolov3 网络结构(基于YoloV5 multi task网络增加分类分支)

yolov3 网络结构(基于YoloV5 multi task网络增加分类分支)

时间2025-05-03 23:22:51分类IT科技浏览3822
导读:前言 YoloV5做目标检测任务,但是当我们需要...

前言

YoloV5做目标检测任务          ,但是当我们需要对图像中的目标进行检测                ,还需要根据图像整体对该图像做分类时      ,我们可以在Yolo-Detect检测头前加一个分类分支     ,这样可以在共享Backbone情况下实现做检测任务                ,又做分类任务           ,下面具体介绍增加分类分支的操作方法

一           、数据准备

如下图     ,yolov5的标签 .txt 文件, 只需在第一行新增类别标签(0                ,1           ,2…),第二行开始是目标框bbox标签

二                、数据输入处修改代码

对标签label增加分类分支

三     、网络修改

四           、模型输出的和标签接上去                ,计算loss

五                、最终可视化结果

原版yolo输出3个检测头                 ,增加后多输出一个分类分支

总结

简单记录一下,因为yolo版本也在不断更新          ,说一没有详细说明                 ,简单对主要修改处标记了      ,如有类似修改需求          。大致参照思路即可                。不能全抄      。

声明:本站所有文章          ,如无特殊说明或标注                ,均为本站原创发布     。任何个人或组织      ,在未征得本站同意时     ,禁止复制     、盗用      、采集                、发布本站内容到任何网站          、书籍等各类媒体平台                。如若本站内容侵犯了原著者的合法权益                ,可联系我们进行处理           。

创心域SEO版权声明:以上内容作者已申请原创保护,未经允许不得转载,侵权必究!授权事宜、对本内容有异议或投诉,敬请联系网站管理员,我们将尽快回复您,谢谢合作!

展开全文READ MORE
基于贪心算法的dfa的程序实现(贪心算法-构造哈夫曼数及生成哈夫曼编码,编程实现) DirectUI 图片(DirectUI)