模型在FloodNet数据集上进行了训练,mIOU在0.83左右,可训练自己的数据集
建议在训练网络的时候将输入的训练集其切分为384x384的小图片后,再来进行训练
模型采用标准的UNet,可以采用如下方式训练你自己的模型
数据集地址可以在train.py中修改为你自己的文件夹
python train.py -- --epochs 20 --batch-size 16 --learning-rate 2e-4 --scale 1 --validation 0.1 --classes 10 --amp
其中--amp为半精度训练
--scale是训练的时候对图片进行缩放,已经裁剪为384x384后就不需要再裁剪了
1