MindSpore数据增强使用

时间:2026-02-21 09:46:23

在MindSpore官网教程中,教程的数据增强部分的使用经验。本人已踩过坑,下面说明一下正确的使用方法。

方法/步骤

首先教程是这样讲述的。

MindSpore数据增强使用

使用方法可参照如下

1、导入所需模块

新加入了Inter模块才可。

from mindspore.dataset.transforms.vision import Inter

import matplotlib.pyplot as plt

import mindspore.dataset.transforms.vision.c_transforms as transforms

2、定义数据路径,此处要注意。在image文件下不可直接放置例如图片的数据,需要再建不同类别的文件夹,目的是将不同类别数据也可以同时处理。(例如在image下创建class1、class2、class3、用以区分数据,路径写到image即可。)

DATA_DIR = "./image"

ds1 = ds.ImageFolderDatasetV2(DATA_DIR, decode=True)

resize_op = transforms.Resize(size=(800,800), interpolation=Inter.LINEAR)

ds2 = ds1.map(input_columns="image", operations=resize_op)

© 2026 一点资料
信息来自网络 所有数据仅供参考
有疑问请联系站长 site.kefu@gmail.com