深度卷积神经网络用于多波段卫星图像的语义分割
准备
从下载3频段和16频段并提取到数据文件夹
通过执行以下命令来安装需求:
$ pip install -r requirements.txt
此外,您需要安装tensorflow或tensorflow-gpu
训练
$ python train.py
争论
描述
选项
--algorithm
训练算法
unet , fcn_densenet , tiramisu , pspnet
--size
补丁大小
整型
--epochs
训练的纪元
整型
--batch
每批样品
整型
--channels
影像频道
3 , 8 , 16
--loss
损失函数
crossentropy , jaccard , dice , cejaccard , cedice
--verbose
打印更多信息
布尔
--noaugment
1