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

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

时间2025-09-19 10:47:40分类IT科技浏览7334
导读:前言 YoloV5做目标检测任务,但是当我们需要...

前言

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

一                、数据准备

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

二                       、数据输入处修改代码

对标签label增加分类分支

三        、网络修改

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

五                       、最终可视化结果

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

总结

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

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

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

展开全文READ MORE
seo教程排名优化网站推广(新的seo网站优化排名 排名)