尧图网络 高端网站定制 · 原创设计
免费咨询热线
400-888-6620
免费获取方案
简要介绍 torchvision.datasets.ImageFolder
torchvision.datasets.ImageFolder专门用来读取按文件夹分类的图像数据集是图像分类任务最常用的自定义数据集类。一、数据集目录强制格式必须遵循下面的层级root/ 类别A/ 图片1.jpg 图片2.png 类别B/ 图片3.jpg ...root数据集根路径传给第一个参数一级子文件夹名称 类别名子文件夹里存放该类别的所有图片ImageFolder 会自动给每个类别分配数字标签0,1,2…例如猫狗分类文件结构规则只能一层类别文件夹不能多层嵌套。二、函数原型ImageFolder( root, transformNone, target_transformNone, loaderdefault_loader, is_valid_fileNone )root(str)数据集根目录路径。transform (callable, 可选)对图像本身做预处理、数据增强。 接收 PIL 图片返回处理后的图片 / Tensor。 示例Resize、随机翻转、转 Tensor、归一化。target_transform (callable, 可选)对标签做变换。 比如把数字标签转成 one‑hot 编码。loader图片读取函数默认default_loader用 PIL 读取图片。一般不用修改。is_valid_file过滤文件自定义哪些文件才视为有效图片。三、返回对象的结构实例化之后得到一个Dataset对象遍历单条样本(image, label)image经过 transform 后的图像张量 / PIL 图label该类别的数字索引int额外自带两个重要属性dataset.classes→ 类别名称列表[cat,dog]dataset.class_to_idx→ 类别→数字映射字典{cat:0, dog:1}四、运行示例from torchvision import datasets, transforms trans transforms.Compose([ transforms.Resize((224,224)), transforms.ToTensor() ]) dataset datasets.ImageFolder(root./train, transformtrans) print(dataset.classes) print(dataset.class_to_idx) img, label dataset[0] print(img.shape, label)五、搭配 DataLoader 使用训练标准写法import torch dataloader torch.utils.data.DataLoader( dataset, batch_sizebatch_size, shuffletrue, num_workersnum_workers, pin_memoryTrue, drop_lastFalse, )六、调用链路示意图流程ImageFolder扫描文件夹自动生成图片路径 标签transform流水线对每张图片做缩放、翻转、转张量、归一化交给DataLoader打包成 batch送入网络训练七、总结ImageFolder自动扫描指定目录下的子文件夹读取图片并生成分类数据集配合 transform 做预处理用于 PyTorch 图像分类训练。
RELATED

相关推荐

CameraLink远距离传输方案:FPGA+GT Transceivers+ Aurora 8B10B光纤链路详解

CameraLink远距离传输方案:FPGA+GT Transceivers+ Aurora 8B10B光纤链路详解

做机器视觉项目的同学应该都懂,CameraLink相机最让人头疼的往往不是价格,而是那根传输线。标准CameraLink线缆有效距离基本上被限制在10米以内,一旦超过这个距离,信号完整性问题就会接踵而至:花屏、闪断、偶发性丢帧&a…

📅 2026/10/9 1:52:18
实战实录(四):任务栈爆了——一个“跑几天才死“的隐形炸弹

实战实录(四):任务栈爆了——一个“跑几天才死“的隐形炸弹

前三篇讲的是"运行时就炸"的问题。这一篇讲最阴险的一类:运行几小时甚至几天才炸。 硬件看起来没坏,代码逻辑看着也没错,但设备在客户现场"随机死机"——这种问题十有八九,是栈。一、现象:一个&qu…

📅 2026/10/5 16:23:58
UniEmployee开源数字员工平台:可审批、可追溯的AI Agent落地实践

UniEmployee开源数字员工平台:可审批、可追溯的AI Agent落地实践

我在开源社区里翻到 UniEmployee 这个项目的时候,挺有感触的。“AI 数字员工”这个概念喊了有两三年了,但大多数产品要么是套壳聊天机器人,要么是只写了白皮书没做成系统。UniEmployee 的切入点不太一样,它把“能干活、能审批、出…

📅 2026/10/4 10:21:50
MORE NEWS

更多资讯

📰

天津宝坻高压电工证含金量高不高?学历不高也能报考

天津宝坻高压电工证含金量高不高?学历不高也能报考 你是不是也担心自己学历不高,根本报不上天津宝坻高压电工证?其实不是这样的,只要你符合基本条件, 含金量…

📰

之前考的证在外省能不能转过来?汕尾市建筑电工证报考保姆级教程

之前考的证在外省能不能转过来?汕尾市建筑电工证报考保姆级教程 之前考的证在外省能不能转过来?这个问题我几乎每天都能在学员群里看到。特别是从外地回来的朋友,拿着其他省份的电工证,想着能不能在汕尾市直接使用,或者是否需要重新报名考试。今天这期【…

📰

职业技能电工证真实经历:外省证怎么转?行业前景如何?

职业技能电工证真实经历:外省证怎么转?行业前景如何? 之前考的证在外省能不能转过来?这个问题我见过太多人问,我自己也经历过,尤其是在鹤壁市政公用工程行业,很多人拿着外省的电工证,到了本地就用不上,要么是不被认可,要么是流程太复杂,耽误项目进…

📰

焊工电工证相片多大?考前押题攻略助你一次过

焊工电工证相片多大?考前押题攻略助你一次过 之前考的证在外省能不能转过来?这是很多老电工、焊工、高处作业人员最关心的问题。尤其是像在驻马店从事水利工程的工友,经常要跨省作业,证书的互通性直接影响工作机会。不过别急,今天不仅给你讲清楚【焊工电…

📰

3个坑别踩!寮步电工证内容网上查询选对机构不被坑

3个坑别踩!寮步电工证内容网上查询选对机构不被坑 不知道去哪报名怕被中介坑?寮步电工证内容网上查询不清晰,找错机构可能白花钱、白费时间。特别是像我们这种在鹤壁做房建工程的,每天接触电、焊、高空作业, 电工证是刚需…

📰

福州如何报考电工证内部消息全公开 电工证怕被坑的看过来

福州如何报考电工证内部消息全公开 电工证怕被坑的看过来 不知道去哪报名怕被中介坑?福州如何报考电工证的流程你了解吗?别急,今天就给你透个 内部消息 ,从报名到考试再到证书补办,一文讲清楚,避免你踩坑。…

TODAY

今日更新

THIS WEEK

本周精选

THIS MONTH

本月热门

读完文章,想聊聊您的网站?

告诉我们您的行业与需求,资深顾问一对一梳理方案与报价,全程免费。

📞 💬