当前位置：网站首页>TAMNet：A loss-balanced multi-task model for simultaneous detection and segmentation

TAMNet：A loss-balanced multi-task model for simultaneous detection and segmentation

2022-08-11 05:23:00 【zhSunw】

在这里插入图片描述
框架采用SSD和FCN作为pipeline.

  • Task-related Attention Module（TAM）：同时考虑两个任务的特征利用注意力机制加权
    在这里插入图片描述
    1.将两个分支的特征相加之后通过注意力模块（CNN）得到加权图
    2.用加权图对两个特征进行加权再与各自原特征相加得到最终特征
    公示表达：F表示之前的特征，M表示加权图，i、c分别表示空间下标和通道下标
    在这里插入图片描述
  • Optimization method
    大部分做法是手动设置权重W：
    在这里插入图片描述
    本文提出了任务t在第i+1步的难度为：
    在这里插入图片描述
    根据每个任务的难度定义奖励（权重）：
    在这里插入图片描述
    在这里插入图片描述
    在这里插入图片描述
    故：在这里插入图片描述
原网站

版权声明
本文为[zhSunw]所创，转载请带上原文链接，感谢
https://blog.csdn.net/ylwhxht/article/details/124479660

边栏推荐

猜你喜欢

随机推荐