实例分割 mask loss异常值,训练过程报错 debug记录

本文记录了在实例分割模型训练过程中遇到的mask loss异常值和验证报错问题的分析与解决过程。首先,通过对比发现mask loss异常可能是由于gt mask在预处理中未进行相应变换导致的,修正预处理后问题得到解决。接着,发现训练过程中的报错发生在验证阶段,原因是数据处理部分存在None值,修正后问题消除。最后,针对目标框消失的问题,检查模型结构和预处理,发现并修复了共享权重、权重初始化以及小目标过滤阈值设置不当等问题。
摘要由CSDN通过智能技术生成

实例分割包含了目标检测,和检测框里的语义分割。

在搭建好一个模型后,训练过程中遇到了mask loss异常值,目标框消失不见,validation过程报错的情况,这里记录一下对这些问题的分析过程和debug记录。

首先分析一下mask loss异常值的问题
mask loss用的是BCE loss。
mask的值很大,几十万。

拿yolact来做一下对比。
它的mask loss只有一万多(后续处理前),那肯定我的值是不对的。

值大了很多,第一直觉是目标框可能没放对位值,导致gt mask不对。

于是选一张图片验证,原图是这样的

请添加图片描述

在yolact的mask loss部分,提取它的gt mask和gt 目标框,

在这里插入图片描述
可以看到图片是经过翻转的,目标框能和mask的目标部分对应上。

而把模型mask也这样提取出来,

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝羽飞鸟

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值