简单的使用AlexNet网络检测光学遥感影像中的船舶
人工智能基础的结课作业(逃
使用卷积神经网络检测光学遥感影像中的船舶,并通过gardio进行可视化,方便使用者查看效果。
代码中,CNN_1是训练过程,CNN_2是展示过程,针对实验用数据集的预训练模型在这里:OneDrive,该模型可以直接在CNN_2.py中加载。
使用了https://opensar.sjtu.edu.cn 的opensarship-1 数据集,对数据集进行预处理使之大小适合网络。
处理后的数据集在这里:OneDrive