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

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

目录

VGG的数据集处理

1.数据的分类

2.对数据集的处理

VGG的分类标签设置

小结


VGG的数据集处理

1.数据的分类

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

2.对数据集的处理

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

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

VGG的分类标签设置

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

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

小结

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

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

相关推荐
d111111111d21 分钟前
STM32得中断服务函数,为什么不能有返回值
笔记·stm32·单片机·嵌入式硬件·学习
阿蒙Amon30 分钟前
JavaScript学习笔记:12.类
javascript·笔记·学习
光影少年34 分钟前
PostgreSQL数据库学习路线
数据库·学习·postgresql
wjykp43 分钟前
part 3神经网络的学习
人工智能·神经网络·学习
阿蒙Amon1 小时前
JavaScript学习笔记:10.集合
javascript·笔记·学习
快撑死的鱼1 小时前
Llama-factory 详细学习笔记:第六章:DPO (直接偏好优化) 实战 (难点)
笔记·学习·llama
d111111111d1 小时前
连续形式PID和离散PID-详情学习-江科大(学习笔记)
笔记·stm32·单片机·嵌入式硬件·学习
四维碎片1 小时前
【Qt】生产者-消费者模式学习笔记
笔记·qt·学习
立志成为大牛的小牛1 小时前
数据结构——五十九、冒泡排序(王道408)
数据结构·学习·程序人生·考研·算法