paddlepaddle框架构建数据集进行分类问题的时候,会发现数据集在构建的过程中不会构建标签(花分类)

问题描述

在做一个paddlepaddle项目的时候,需要使用神经网络对他进行分类,数据集的结构如下图,这时候我们可以使用常用dataset方法对数据集进行构建。

这时候我们就会发现一个问题,就是这个矿建不是构建标签,也就是说不能将该文件内的图片分配类别,这和torch中的dataset构建是不一样的。那么这时候,就需要我们手动的去构建这个,这时候,我们自定义类,代码如下。

python 复制代码
class CustomImageFolder(DatasetFolder):
    def __init__(self, root, transform=None, ):
        super(CustomImageFolder, self).__init__(root, transform=transform)

    def __getitem__(self, idx):
        img, label = super(CustomImageFolder, self).__getitem__(idx)

        # 添加图像增强
        img = Resize((224, 224))(img)
        img = RandomHorizontalFlip()(img)
        img = ColorJitter(brightness=0.2, contrast=0.2, saturation=0.2, hue=0.1)(img)
        img = ToTensor()(img)
        img = paddle.transpose(img, perm=[2, 0, 1])

        return img, label

# 加载数据集并应用自定义转换
dataset = CustomImageFolder(root=data_dir, transform=None)

这时候返回的数据集就会自带类别标签了,可以将dataset[0]打印出来看看结果。

相关推荐
Moniane1 小时前
A2A+MCP构建智能体协作生态:下一代分布式人工智能架构解析
人工智能·分布式·架构
sendnews2 小时前
红松小课首次亮相北京老博会,四大业务矩阵赋能退休生活提质升级
人工智能·物联网
停停的茶2 小时前
深度学习——图像分割
人工智能·深度学习
MIXLLRED3 小时前
自动驾驶技术全景解析:从感知、决策到控制的演进与挑战
人工智能·机器学习·自动驾驶
金融Tech趋势派3 小时前
企业微信AI SCRM推荐:从技术适配与场景功能实践进行评估
大数据·人工智能
Wnq100723 小时前
AI 在法律咨询服务中的革命性变化:技术赋能与生态重构
人工智能·职场和发展·重构·分类·数据分析·全文检索·创业创新
茶杯6753 小时前
极睿iClip易视频:2025年AI混剪领域的革新工具,重构电商内容生产逻辑
人工智能
一点一木3 小时前
🚀 2025 年 10 月 GitHub 十大热门项目排行榜 🔥
前端·人工智能·github
湘-枫叶情缘3 小时前
程序与工业:从附庸到共生,在AI浪潮下的高维重构
人工智能·重构