参考自:https://blog.csdn.net/u012931582/article/details/70215756
https://github.com/zhixuhao/unet
https://blog.csdn.net/py_yangh/article/details/83003972
文章代码来自:https://blog.csdn.net/awyyauqpmy/article/details/79290710
作者的github:https://github.com/decouples/Unet
1、libtiff 的安装
pip install libtiff 安装失败,只能在官网下载 whl 文件手动安装
安装过程见:https://blog.csdn.net/weixin_42338058/article/details/89883411
2、data 文件路径
尝试了无数次原作者的文件夹结构都不行,只好自己修改文件夹路径
文件夹结构如下:
--aug_label #数据增强后的标签
--0
0.tif
--1
1.tif
--2
2.tif
.
.
--aug_merge #数据增强后,增强数据和标签数据在同一张图中,这里应该是伪彩色图
--0
0.tif
--1
1.tif
--2
2.tif
.
.
--aug_train #数据增强后,split合并的训练数据和标签
--0
0.tif
--1
1.tif
-- 2
2.tif
.
.
--images #训练数据,标签及测试数据所在文件夹路径
--train
0.tif
1.tif
.
.
--test
0.tif
1.tif
.
.
--test
0.tif
1.tif
.
.
--merge
0.tif
1.tif
.
.
--npydata
imgs_mask_train.npy
imgs_test.npy
imgs_train.npy
3.训练次数
不知道是不是数据量太少,我没用增强后的数据训练,只有30张训练数据,训练10次的acc是0.5左右,20次是0.89的样子,30次是0.69,50次也是0.69,不能增加太多次,未解决该问题~~
PS:我的github代码已经修改并跑通,大概89%的准确率,欢迎加星啊~嘻嘻,下面是原图和分割结果