主要策略:
- 将原始的遥感图像裁成大小为(256x256)的图片块,裁剪的方法为随机采样,并进行数据扩增
- 搭建Deeplab-v3模型,使用预训练的 resnet-v2-50 迁移学习
- 完整的训练测试程序,使用 tensorboard 监控模型训练
- 多尺度拼接预测,提升模型
- 后处理优化,比如消除预测图片拼接痕迹
- 使用更好的骨干网络,如 Xception
Deeplabv3模型和resnet-v2-50的关系,为什么可以用resnet-v2-50做迁移学习
迁移学习怎么做?
ValueError: Variable resnet_v2_50/conv1/weights/Adam/ already exists, disallowed. Did you mean to set reuse=True or reuse=tf.AUTO_REUSE in VarScope? Originally defined at:
https://github.com/anxiangsir/deeplabv3-Tensorflow
Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
CUDA版本问题。