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

bubbliiiing / faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库，可以利用voc数据集格式的数据进行训练。

MIT License

1.56k stars 352 forks source link

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

Open Scienthusiasts opened 1 year ago

Scienthusiasts commented 1 year ago

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

Scienthusiasts commented 1 year ago

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

Scienthusiasts commented 1 year ago

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

Scienthusiasts commented 1 year ago

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

Scienthusiasts commented 1 year ago

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