yolov3 网络结构(基于YoloV5 multi task网络增加分类分支)
导读:前言 YoloV5做目标检测任务,但是当我们需要...
前言
YoloV5做目标检测任务 ,但是当我们需要对图像中的目标进行检测 ,还需要根据图像整体对该图像做分类时 ,我们可以在Yolo-Detect检测头前加一个分类分支 ,这样可以在共享Backbone情况下实现做检测任务 ,又做分类任务 ,下面具体介绍增加分类分支的操作方法
一 、数据准备
如下图 ,yolov5的标签 .txt 文件, 只需在第一行新增类别标签(0 ,1 ,2…),第二行开始是目标框bbox标签
二 、数据输入处修改代码
对标签label增加分类分支
三、网络修改
四 、模型输出的和标签接上去 ,计算loss
五 、最终可视化结果
原版yolo输出3个检测头 ,增加后多输出一个分类分支
总结
简单记录一下,因为yolo版本也在不断更新 ,说一没有详细说明 ,简单对主要修改处标记了,如有类似修改需求 。大致参照思路即可 。不能全抄 。
声明:本站所有文章 ,如无特殊说明或标注 ,均为本站原创发布 。任何个人或组织 ,在未征得本站同意时 ,禁止复制 、盗用 、采集 、发布本站内容到任何网站 、书籍等各类媒体平台 。如若本站内容侵犯了原著者的合法权益 ,可联系我们进行处理 。
创心域SEO版权声明:以上内容作者已申请原创保护,未经允许不得转载,侵权必究!授权事宜、对本内容有异议或投诉,敬请联系网站管理员,我们将尽快回复您,谢谢合作!