在增加mask分支后,mask loss权重设置为0,训练2k次,检测效果下降严重:

mask-rcnn中检测分支性能下降调查

去除mask分支后,训练2k次,检测效果恢复:

mask-rcnn中检测分支性能下降调查

目前正在调查是mask分支具体哪部分引起的问题,希望尽快找到问题症结所在。


经调查发现是如下分割分支导致的检测功能异常问题,可能反向传播的梯度过大:

mask-rcnn中检测分支性能下降调查

该问题已经被解决了,采用如下网络结构,这里主要起作用的是bn等操作:

mask-rcnn中检测分支性能下降调查

最终增加了一个roi_pooling使得与检测分支分开,训练速度更快:

mask-rcnn中检测分支性能下降调查

相关文章:

  • 2022-12-23
  • 2021-07-21
  • 2021-06-10
  • 2021-12-06
  • 2021-09-08
  • 2021-07-14
猜你喜欢
  • 2022-01-01
  • 2021-04-03
  • 2022-12-23
  • 2021-11-30
  • 2021-05-15
  • 2021-08-06
  • 2021-08-21
相关资源
相似解决方案