【pytorch框架移植】【图像增强处理】torchvision.transforms.Lambda变换无对应API

问题描述：

【功能模块】

torchvison.transform.Lambda()与mindspore框架的transforms模块找不到对应

【操作步骤&问题现象】

torchvision.transforms.Lambda()对应的mindspore变换。

在做pytorch与mindspore框架迁移的时候，找不到对应的用户自定义Lambda函数操作变换。

【截图信息】

解答：

请参考一下这个用法：

https://www.mindspore.cn/docs/programming_guide/zh-CN/r1.6/pipeline_common.html#map

import numpy as np

import mindspore.dataset as ds

def generator_func():

    for i in range(5):

        yield (np.array([i, i+1, i+2]),)

def pyfunc(x):

    return x*2

dataset = ds.GeneratorDataset(generator_func, ["data"])

dataset = dataset.map(operations=pyfunc, input_columns=["data"])

for data in dataset.create_dict_iterator():

    print(data)

或者参考modelzoo中的写法

https://gitee.com/mindspore/models/blob/master/official/cv/ssd/src/dataset.py