/D-linknet

实现unet,以及改进版,deeplapv3+,等网络结构,同时可以快速训练,识别进行语义分割!

Primary LanguagePython

dliknet、unet、deeplabv3+

整体框架为北京邮电大学的道路识别比赛,在此基础上学习,修改,添加了deeplabv3+网络,可以实现遥感数据的数据集制作,网络切换,模型训练,以及结果输出。

主体代码为train.py文件,设置好40行训练数据,和标签的路径,选择56行自己需要的网络结构,执行即可开始训练。

同时提供ToolGUI.py工具可以将遥感大图裁剪为256*256训练集,自动生成dataset文件夹。

image

同时添加了网络训练过程可视化,可实时查看训练过程,代码中已经注释,需要自己选择。

image

image

image

训练完毕,调用predict_best.py设置好mian后的参数,执行预测,预测采用只取中间1/4,经过测试,这样效果比较好。

imageimageimage

imageimageimage

imageimageimage

分别为预测结果、影像、和标签,可以看出来网络训练不好,还是会出现棋盘现象,训练时间1080T 20个epoch,样本总数为30000张256*256.

技术交流联系方式:mrwanghongji@qq.com