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

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

目录

VGG的数据集处理

1.数据的分类

2.对数据集的处理

VGG的分类标签设置

小结


VGG的数据集处理

1.数据的分类

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

2.对数据集的处理

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

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

VGG的分类标签设置

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

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

小结

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

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

相关推荐
饮长安千年月2 小时前
Linksys WRT54G路由器溢出漏洞分析–运行环境修复
网络·物联网·学习·安全·机器学习
红花与香菇2____2 小时前
【学习笔记】Cadence电子设计全流程(二)原理图库的创建与设计(上)
笔记·嵌入式硬件·学习·pcb设计·cadence·pcb工艺
一天八小时4 小时前
Docker学习进阶
学习·docker·容器
前端没钱4 小时前
前端需要学习 Docker 吗?
前端·学习·docker
拥有一颗学徒的心5 小时前
鸿蒙第三方库MMKV源码学习笔记
笔记·学习·性能优化·harmonyos
车端域控测试工程师5 小时前
【ISO 14229-1:2023 UDS诊断(ECU复位0x11服务)测试用例CAPL代码全解析⑰】
经验分享·学习·汽车·测试用例·capl
车端域控测试工程师5 小时前
【ISO 14229-1:2023 UDS诊断(ECU复位0x11服务)测试用例CAPL代码全解析⑪】
经验分享·学习·汽车·测试用例·capl
charlie1145141918 小时前
(萌新入门)如何从起步阶段开始学习STM32 —— 0.碎碎念
c语言·stm32·单片机·嵌入式硬件·学习·教程
网安Ruler10 小时前
泷羽Sec-黑客基础之html(超文本标记语言)
前端·学习·网络安全·html
啥也不会的菜鸟·11 小时前
Redis7——基础篇(五)
redis·学习·缓存