学习基于pytorch的VGG图像分类 day5

注:本系列博客在于汇总CSDN的精华帖,类似自用笔记,不做学习交流,方便以后的复习回顾,博文中的引用都注明出处,并点赞收藏原博主.

目录

VGG的数据集处理

1.数据的分类

2.对数据集的处理

VGG的分类标签设置

小结


VGG的数据集处理

1.数据的分类

对于收集到的数据分成三类,分别用于测试,训练和验证,训练和验证集的大小差不多在8:3左右,放在同一文件夹下。

2.对数据集的处理

将数据进行分类的放在对应的文件夹下。

同时对所有数据生成一个csv文件,记录各个数据的路径和分类标签,具体生成方法自行上网查找,这里不在一一赘述。

VGG的分类标签设置

在根目录写一个json文件,用来表示分类编号,具体如图所示。

对于如何写一个json文件,可以新建一个记事本文件,写完后保存格式改为json即可。

小结

1.对于数据集的处理非常重要,直接影响到能否使模型成功训练

2.数据集可以从网上下载,也可以使用爬虫拔取数据

相关推荐
xw-busy-code29 分钟前
Prettier 学习笔记
javascript·笔记·学习·prettier
毕设源码-郭学长35 分钟前
【开题答辩全过程】以 课程学习过程性评价系统为例,包含答辩的问题和答案
学习
酸奶乳酪35 分钟前
IIC学习笔记
笔记·单片机·学习
小陈phd1 小时前
系统架构师学习笔记(二)——计算机体系结构之指令系统
笔记·学习·系统架构
吃杠碰小鸡2 小时前
前端 IndexedDB 完全指南
学习
问道飞鱼2 小时前
【大模型学习】LangGraph 深度解析:定义、功能、原理与实践
数据库·学习·大模型·工作流
烤麻辣烫3 小时前
I/O流 基础流
java·开发语言·学习·intellij-idea
云边散步3 小时前
godot2D游戏教程系列二(22)
笔记·学习·游戏
jincheng_3 小时前
软件设计师上午题|9模块极速背诵版
学习
Schengshuo3 小时前
Spring学习——新建module模块
java·学习·spring