CCF卫星影像的AI分类与识别提供的数据集初赛复赛训练集,一共五张卫星遥感影像
- 使用模型为deeplabv_3,使用预训练好的resnet_v2_50 fine-tuning
- 这里放的预测图是单模型跑出来的结果,多模型融合肯定会使得结果变得更好
- 没有针对道路类别进行过采样,导致道路类别分的不好
- 采样方法为随机采样,划窗采样
- 旋转90°,180°,270°
- 左右翻转,上下翻转
- 将百度云中的数据集文件夹dataset下载并存放到项目主目录下
- 运行 proprecess.py 生成训练集 时间稍长,需要等待
- 运行 train.py 开始训练