文档代码运行报错

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版权协议,转载请附上原文出处链接和本声明。