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

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

目录

VGG的数据集处理

1.数据的分类

2.对数据集的处理

VGG的分类标签设置

小结


VGG的数据集处理

1.数据的分类

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

2.对数据集的处理

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

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

VGG的分类标签设置

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

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

小结

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

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

相关推荐
Broken Arrows6 小时前
Linux学习——管理网络安全(二十一)
linux·学习·web安全
今天也要学习吖6 小时前
谷歌nano banana官方Prompt模板发布,解锁六大图像生成风格
人工智能·学习·ai·prompt·nano banana·谷歌ai
雁于飞7 小时前
vscode中使用git、githup的基操
笔记·git·vscode·学习·elasticsearch·gitee·github
rannn_1117 小时前
【Javaweb学习|实训总结|Week1】html基础,CSS(选择器、常用样式、盒子模型、弹性盒布局、CSS定位、动画),js(基本类型、运算符典例)
css·笔记·学习·html
wanzhong23339 小时前
ArcGIS学习-20 实战-地形研究
学习
wanzhong23339 小时前
ArcGIS学习-20 实战-县域水文分析
学习·arcgis
小马学嵌入式~9 小时前
嵌入式 SQLite 数据库开发笔记
linux·c语言·数据库·笔记·sql·学习·sqlite
索迪迈科技9 小时前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法
武文斌7710 小时前
arm启动代码总结
arm开发·嵌入式硬件·学习