在增加mask分支后,mask loss权重设置为0,训练2k次,检测效果下降严重:
去除mask分支后,训练2k次,检测效果恢复:
目前正在调查是mask分支具体哪部分引起的问题,希望尽快找到问题症结所在。
经调查发现是如下分割分支导致的检测功能异常问题,可能反向传播的梯度过大:
该问题已经被解决了,采用如下网络结构,这里主要起作用的是bn等操作:
最终增加了一个roi_pooling使得与检测分支分开,训练速度更快:
在增加mask分支后,mask loss权重设置为0,训练2k次,检测效果下降严重:
去除mask分支后,训练2k次,检测效果恢复:
目前正在调查是mask分支具体哪部分引起的问题,希望尽快找到问题症结所在。
经调查发现是如下分割分支导致的检测功能异常问题,可能反向传播的梯度过大:
该问题已经被解决了,采用如下网络结构,这里主要起作用的是bn等操作:
最终增加了一个roi_pooling使得与检测分支分开,训练速度更快:
相关文章: