如何使用mxnet加载灰度png文件？

Question

我想使用mxnet Image API加载灰度png文件。 但Image API成为3通道RGB数据。 如何将3channel数据转换为1channel数据？

Answer 1

您可以使用imdecode函数并将标志设置为0。 https://github.com/apache/incubator-mxnet/blob/master/python/mxnet/image/image.py#L85

这将要求您安装opencv并使用USE_OPENCV = 1

编译mxnet