
印度松树高光谱图像的简单分类:基于CNN和Keras的分类方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了一种使用卷积神经网络(CNN)及Keras框架对印度松树高光谱图像进行有效分类的方法,为植被监测提供技术支持。
Indian_pines_classification的运行环境为Anaconda 3.6.4, 其中Python版本为3.6.4,Keras版本为2.1.5,TensorFlow版本为1.3.0作为keras后端,numpy版本为1.14.2,notebook版本为5.4.1,ipython版本为6.3.1,spectral工具包的版本是0.19。
代码结构包括两个主要文件:dataset.ipynb和train.ipynb。在dataset.ipynb中使用spectral工具包读取数据集,并对数据进行预处理,这其中包括训练测试集分割、序列化以及PCA变换等操作,最后将这些处理后的数据以npy格式保存至predata文件夹。
而在train.ipynb中,则利用keras构建卷积神经网络模型,使用之前经过处理的数据集来完成模型的训练。在该过程中采用随机梯度下降法(SGD)作为优化算法,并选择多分类对数损失函数categorical_crossentropy作为损失函数。此外,在训练阶段还应用了ReduceLROnPlateau机制。
全部评论 (0)
还没有任何评论哟~


