Advertisement

0009期鱼类识别-包含数据集与代码 0009期鱼类识别-包含数据集与代码

阅读量:

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

获取该代码后,会发现一个名为requirement.txt的文件,其中详细说明了环境配置的具体步骤。

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

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

本代码对数据集实施了预处理操作,具体包括在图片较短边处添加灰色边缘,以将图像转换为正方形格式(若原始图像本身即为正方形,则无需添加灰色边缘),同时通过调整旋转角度的方式,对数据集进行扩充与增强。

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

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

![](https://cdl.itadn.

全部评论 (0)

还没有任何评论哟~