维度不正确"/>
【mindspore】【dataset.GeneratorDataset】数据处理后的维度不正确
dataset.GeneratorDataset 处理数据的维度不正确
【操作步骤&问题现象】
1、导入自定义的数据并进行处理
2、进行模型训练
问题现象:
就是本人希望输入数据的维度是[32,3,224,224]
但是经过上述的操作后维度为[32,1,3,224,224]
不太清楚中间发生了什么错误,是API没有用对还是说数据集代码写的有问题,求帮忙指点一下。
【截图信息】
【日志信息】(可选,上传日志内容或者附件)
1、数据集代码
2、训练代码
怀疑是数据处理逻辑的问题,python代码多写了一维。
更多推荐
【mindspore】【dataset.GeneratorDataset】数据处理后的维度不正确
发布评论