Advertisement

深度学习识别国际象棋游戏(附带数据集和代码)

阅读量:

本代码的运行环境为Python与PyTorch框架。

在获取该代码后,会发现一个名为requirement.txt的文件,其中详细说明了相关环境的安装方式。

关于数据集的信息,当下载完该资源后,界面将呈现如下内容:

本次识别任务所涉及的各类图像数据均存储于数据集文件夹中。

本代码对所使用的数据集实施了预处理操作,具体包含两个方面:一是通过在图像较短边处添加灰色边缘,使图像尺寸调整为正方形(若原始图像本身为正方形则无需此操作),二是对图像进行角度旋转处理,以此实现数据集的扩充与增强。

执行01数据集文本制作.py脚本后,系统将自动获取数据集内各分类文件夹中图像的存储路径及其对应标签信息。

启动02深度学习模型训练.py程序后,系统会读取txt文件中所记录的训练样本与验证样本,完成模型训练过程。待训练完成后,生成的模型将被保存至本地存储位置。

![](https://cdl.it

全部评论 (0)

还没有任何评论哟~