Как использовать mxnet для загрузки файла png в градациях серого?

Я хочу использовать mxnet Image API для загрузки файла png в градациях серого. Но Image API стал 3-канальными данными RGB. Как преобразовать 3-канальные данные в 1-канальные?


mxnet
person user2850652    schedule 15.08.2017    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Вы можете использовать функцию imdecode и установить флаг с 0. https://github.com/apache/incubator-mxnet/blob/master/python/mxnet/image/image.py#L85

Для этого вам потребуется установить opencv и скомпилировать mxnet с USE_OPENCV=1.

person user3782250    schedule 15.08.2017
comment
Могу ли я использовать imgdecode с mxnet.image.ImageDetIter с параметром aug_list? - person user2850652; 16.08.2017