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