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

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

目录

VGG的数据集处理

1.数据的分类

2.对数据集的处理

VGG的分类标签设置

小结


VGG的数据集处理

1.数据的分类

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

2.对数据集的处理

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

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

VGG的分类标签设置

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

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

小结

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

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

相关推荐
坚持是一种态度10 小时前
软考高级架构师选择题知识点高频100
学习·架构师·软考·考试
yxc_inspire10 小时前
25年CCPC福建邀请赛补题
学习·算法
Slow菜鸟10 小时前
Skill 学习篇(十)| 编排框架 · 五大编排框架 · 全方位决策指南
学习
倒霉熊dd10 小时前
Python学习(第一部分 语法与数据结构/核心基础)
大数据·python·学习·pip
星夜夏空9911 小时前
STM32单片机学习(3)——前置知识学习
stm32·单片机·学习
Shadow(⊙o⊙)12 小时前
初识Qt+经典方式实现hello world!的交互
开发语言·c++·后端·qt·学习
星夜夏空9912 小时前
STM32单片机学习(5) —— STM32的一些名词解释
stm32·单片机·学习
拾21412 小时前
小论文中使用他人文章图表的权限获取
学习
GEO从入门到精通12 小时前
GEO课程的学习路径应该怎么规划?
人工智能·学习
炽烈小老头13 小时前
【 每天学习一点算法 2026/05/08】最小覆盖子串
学习·算法