https://www.mindspore.cn/docs/zh-CN/r1.7/note/api_mapping/tensorflow_diff/grayscale_to_rgb.html
【操作步骤&问题现象】
文档代码
# The following implements ConvertColor with MindSpore.
import numpy as np
import mindspore.dataset as ds
from mindspore.dataset.vision import ConvertMode
image = np.random.random((28, 28, 1))
result = ds.vision.c_transforms.ConvertColor(ConvertMode.COLOR_GRAY2RGB)(image)
print(result.shape)
# (28, 28, 3)
【日志信息】(可选,上传日志内容或者附件)
文档上面复制下来的代码运行报错 RuntimeError: Unexpected error. ConvertColor: number of channels of image should be 3 or 4, but got:1
mindspore_cpu 1.7.0
复现了一下,的确有问题,看了一下源码,源码在检查通道上的时候有bug。
问题已经定位到,需要修改:image = np.random.random((28, 28, 1)).astype(np.float32);同时我们内部的算子有误正在修改。
版权声明:本文为xi_xiyu原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。