bubbliiiing/faster-rcnn-pytorch

在更换Backbone后是否需要冻结BN层

Scienthusiasts opened this issue · 4 comments

Scienthusiasts commented

B导您好！我尝试重写DataLoader(改写成读取COCO格式数据集)，并使用Albunmentation做数据增强，但是读取后发现网络的输出结果不正常，以下是backbone层的输出结果，想请教一下可能会是什么原因：
image
这是我重写之后的FRCNNDataLoader附件：
FRCNNDataLoader.zip
训练时需要在train.py里更改以下内容：
image

Scienthusiasts commented

我目前怀疑是使用Albunmentation做数据增强的原因

Scienthusiasts commented

已找到原因：coco格式为了匹配代码还得把box的坐标转成VOC格式

  • 👍1
Scienthusiasts commented

之后通过排除问题，依然没有消除网络输出不正常的问题，目前已经排除了重写的Dataloader的问题，现在问题可能出在重写的网络上

Scienthusiasts commented

发现：取消冻结BN层网络更容易收敛，否则就会出现该现象，
原因：有待进一步考察