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

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

目录

VGG的数据集处理

1.数据的分类

2.对数据集的处理

VGG的分类标签设置

小结


VGG的数据集处理

1.数据的分类

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

2.对数据集的处理

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

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

VGG的分类标签设置

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

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

小结

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

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

相关推荐
盐焗西兰花8 分钟前
鸿蒙学习实战之路 - 网络重连最佳实践
网络·学习·harmonyos
酒尘&4 小时前
JS数组不止Array!索引集合类全面解析
开发语言·前端·javascript·学习·js
冬夜戏雪4 小时前
【java学习日记】【2025.12.7】【7/60】
java·开发语言·学习
wubba lubba dub dub7505 小时前
第二十八周 学习周报
学习
思成不止于此5 小时前
MySQL 查询实战(三):排序与综合练习
数据库·笔记·学习·mysql
QiZhang | UESTC5 小时前
学习日记day42
学习
深海潜水员5 小时前
OpenGL 学习笔记 第一章:绘制一个窗口
c++·笔记·学习·图形渲染·opengl
大、男人7 小时前
DeepAgent学习
人工智能·学习
测试人社区—66797 小时前
提升测试覆盖率的有效手段剖析
人工智能·学习·flutter·ui·自动化·测试覆盖率
崇山峻岭之间7 小时前
C++ Prime Plus 学习笔记026
c++·笔记·学习