0013期使用CNN训练猫狗数据集识别(附带代码)
发布时间
阅读量：
阅读量
本代码的运行环境为Python与PyTorch框架。
在获取该代码后，会发现一个名为requirement.txt的文件，其中详细说明了环境配置的具体步骤。
关于数据集的相关信息，当下载完该资源后，所呈现的界面如下：
本次识别任务所涉及的各类图像数据均存储于数据集文件夹中。
本代码对所使用的数据集实施了预处理操作，具体包括在图像较短边处添加灰色边缘以实现正方形格式转换（若原始图像本身为正方形则无需进行此操作），同时通过调整旋转角度的方式对图像进行变换，从而达到扩充和增强数据集的目的。
执行01数据集文本制作.py脚本后，系统将自动获取数据集中各分类文件夹内所有图像的存储路径及其对应类别标签信息。
启动02深度学习模型训练.py程序后，系统会读取txt文件中所记录的训练样本与验证样本数据，完成模型训练过程。训练结束后，生成的模型文件将会被保存至本地存储位置。
还没有任何评论哟~