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

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

目录

VGG的数据集处理

1.数据的分类

2.对数据集的处理

VGG的分类标签设置

小结


VGG的数据集处理

1.数据的分类

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

2.对数据集的处理

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

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

VGG的分类标签设置

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

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

小结

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

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

相关推荐
虾球xz9 分钟前
游戏引擎学习第252天:允许编辑调试值
c++·学习·游戏引擎
jackson凌23 分钟前
【Java学习笔记】递归
java·笔记·学习
DisonTangor34 分钟前
Meta 推出 WebSSL 模型:探索 AI 无语言视觉学习,纯图训练媲美 OpenAI CLIP
人工智能·学习·计算机视觉
s_little_monster2 小时前
【Linux】线程池和线程补充内容
linux·运维·服务器·c++·笔记·学习·学习方法
Timmer丿2 小时前
kafka学习笔记(四、生产者(客户端)深入研究(二)——消费者协调器与_consumer_offsets剖析)
笔记·学习·kafka
李匠20243 小时前
C++负载均衡远程调用学习之Reactor事件触发机制
c++·学习
冰茶_4 小时前
WPF之Image控件详解
学习·microsoft·微软·c#·wpf·wpf控件
搏博4 小时前
结构模式识别理论与方法
人工智能·深度学习·学习·算法·机器学习
z35026037064 小时前
K8S学习笔记01
笔记·学习·kubernetes
小猪佩奇TONY5 小时前
OpenGL-ES 学习(12) ---- VBO EBO VAO
学习