首页 > 其他分享 >关于多任务学习MTL的实现步骤记录

关于多任务学习MTL的实现步骤记录

时间:2022-09-27 12:14:17浏览次数:49  
标签:步骤 分类 损失 多任务 MTL 检测 类别 预处理

拿任务来讲解:

假设我目前已经有了一个目标检测功能,检测物体a,现在我想判断这个a是不是真实场景下的,需要对整图再加个判断,即二分类。这样的需求其实就是典型的多任务学习,即检测+分类

拿到这样的任务,目前经过实际操作已知如下有效的训练方法。

  1. 首先,训练数据要保证类别均衡,包括两个方面,检测就不用说了,最好是各个类别均衡,其次是分类方面,检测类别的实拍和非实拍也要均衡
  2. 然后数据预处理,假设batch是32,那么我们需要从分类数据中取出16张,从检测数据中取出16张,然后用各自的预处理方式进行预处理,最后组合成batch-32,送到模型中
  3. 计算损失,此时需要在backbone上加入一个分类头,以yolov5为例,这个分类头可以加到三个输出中的任何一个当中都可以,一般情况下加到最后一个输出上
  4. 在损失计算上,根据分类和检测的图片位置,分别计算相应的分类损失和检测损失,然后相加后反向传播,至此,多任务的整体训练流程就结束

 

标签:步骤,分类,损失,多任务,MTL,检测,类别,预处理
From: https://www.cnblogs.com/chentiao/p/16734093.html

相关文章

  • RT-THREAD nano移植步骤
    1.拷贝所有文件,libcpu,bsp文件夹中只需要拷贝对应M4/M3文件  2.rtthread_startup函数芯片外设初始化,systick初始化。芯片库自带的systick中断入口函数去掉,因为rt-thre......
  • Junit测试概述和Junit使用步骤
    Junit测试概述测试分类:黑盒测试:不需要写代码。给输入值,看程序是否能输出期望的值白盒测试:需要些代码的。关注程序具体的执行流程Junit使用步......
  • gitee上提交代码的步骤
    0.预备步骤:要再提交的目标master文件中fork一个分支,才能进行后续的步骤。1.首先要从gitee代码仓库中拉取代码下来,pull,其次要在本地新建一个文件夹将要上传的代码。使用git......
  • Zookeeper升级新版本步骤
    Zookeeper升级新版本步骤一、需求由于安全漏洞扫描,需要对zookeeper进行升级。从3.3.6升级至3.4.9。二、升级步骤去官方下载zookeeper的包。wgethtt......
  • 软件测试的一般步骤
    11.7软件测试的一般步骤除非是测试一个小程序,否则一开始就把整个系统作为一个单独的实体来测试是不现实的。与开发过程类似,测试过程也必须分步骤进行,后一个步骤在逻辑上......
  • 海康摄像机使用GB28181接入SRS服务器的搭建步骤---封装docker镜像运行容器的方式,本篇
    使用第三方提供的镜像说明:这个是我在网上找的,没有实际验证及使用过,具体效果未知,看下文,我采用的是自己依据官方基础镜像进一步二次封装的镜像地址:https://hub.docker.com/......
  • 海康摄像机使用GB28181接入SRS服务器的搭建步骤---源码安装的方式
    下载代码地址:https://github.com/ossrs/srs-gb28181https://github.com/ossrs/srs-gb28181.git注意:使用的是含有gb28181的GitHub仓库地址。若是使用其他的仓库地址,比如......
  • 数字化转型需要经历哪些步骤?
    数字化转型大致分为下列五个步骤:1、首当其冲的是战略上的转型!客观来讲,数字化转型实际上是“一把手”的一场自我革命之旅,“一把手”要有自我否定的勇气,摒弃路径依赖,重塑......
  • 二分查找步骤及问题总结
    二分查找参数:有序数组arr(这里按升序来讲),待搜索的值target步骤定义左边界left和有边界right获取中间索引(整数)mid=(left+right)/2,注意:js只有小数,mid需要再取整......
  • Anaconda功能、优点、安装步骤(安装视频)
    目录介绍功能(包和环境的管理器)优点(省时省心)下载地址安装教程要点conda的常见命令查询完整帮助文件管理conda和anaconda管理环境包管理其他​介绍Anaconda是专注于数据......