MindSpore数据增强使用

2025-12-19 11:19:25

在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)

声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
猜你喜欢