/faster_rcnn_in_tf2_keras

tf2-keras implement faster-rcnn

Primary LanguagePython

faster rcnn in tf2-keras

bilibili视频讲解

Watch the video

测试效果 Faster-RCNN

  • VOC2012

  • bd100k

一些训练要点

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