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

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

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

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

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


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

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

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

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

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

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

相关文章: