版权声明:本文为博主原创文章,转载请注明出处。 https://blog.csdn.net/nima1994/article/details/84031759
代码位置:https://github.com/lsh1994/keras-segmentation
实验环境
win10;python3.6 .7 ;
项目 | Value |
---|---|
keras | 2.2.4 |
tensorflow-gpu | 1.10.0 |
实验参考:https://github.com/divamgupta/image-segmentation-keras
实验数据:https://drive.google.com/file/d/0B0d9ZiqAgFkiOHR1NTJhWVJMNEU/view
实验数据或者: https://github.com/alexgkendall/SegNet-Tutorial/tree/master/CamVid
文件结构
python visualizeDataset.py
可视化样本:
python train.py
:训练
python predict.py
:预测
进展
目前完成了FCN32、FCN8。
FCN32
训练曲线:
预测结果:
FCN8
文章过长,参考下篇。
扫描二维码关注公众号,回复:
4876929 查看本文章