faster rcnn in tf2-keras
bilibili视频讲解
测试效果 Faster-RCNN
- VOC2012
- bd100k
一些训练要点
- faster-rcnn样本平衡原则:
faster-rcnn即使最后对正负样本做了限制, 但对于单目标小目标的样本来说还是会导致训练不平衡, 故可在最后计算loss处再平衡一次.
- loss取舍问题:
由于rcnn系列loss来源多个, 综合起来最后loss可能过大, 可观察后调整, 开始loss在0.5-1.0较为合适, 倘若loss一直比较大, 训练容易导致奔溃loss爆炸.