BiSeNet训练总结笔记
针对BiSeNet语义分割模型,利用开源的pytorch项目,进行了训练尝试。主要是利用不同的head network(res18和res101),结合不同的优化方法(rmsprop和sgd),在不同batch下(1,2,4,8)进行Camvid数据集的训练。
训练结束以后,利用如下图片和真实标签进行测试和对比:
学习率poly更新策略
其中初始学习率为0.01,采用poly_lr_scheduler策略,按epoch不断下降。
(图中的iteration应该为epoch)
Res18
rmsprop优化器
batch1