
利用Python深度学习进行岩石识别-附带数据集.zip
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本资源提供基于Python深度学习技术进行岩石图像自动分类的方法及代码,包含用于训练和测试的岩石图像数据集,适合科研与教学使用。
本代码基于Python PyTorch环境编写。
下载后,请查看文件中的requirement.txt以了解如何安装所需环境;您也可以参考相关文档进行设置。
配置好环境之后,
请依次运行以下三个脚本:
1. 01数据集文本生成制作.py
2. 02深度学习模型训练.py
3. 03pyqt_ui界面.py
这些代码对数据集进行了预处理,包括通过在较短边增加灰度背景以使图片变为正方形(如果原图已经是正方形则不会进行此操作),以及旋转角度来增强数据集。
运行第一个脚本时,会读取每个类别文件中的图像路径及对应的标签;
第二个脚本将使用训练和验证集合的txt文本开始模型训练,并在完成后保存至本地。
全部评论 (0)
还没有任何评论哟~


