Есть ли способ импортировать изображения MNIST только с 0 и 1?

Я только начинаю работать с тензорным потоком и хочу что-то протестировать только на 0 и 1 из изображений MNIST. Есть ли способ импортировать только эти изображения?


person Kes    schedule 25.02.2017    source источник


Ответы (1)


Предполагая, что вы используете

from tensorflow.examples.tutorials.mnist import input_data

Нет, в этом файле нет функции или аргумента... Что вы можете сделать, так это загрузить все данные и выбрать только единицы и нули.

person rmeertens    schedule 25.02.2017
comment
Извините, что неясно, но да, я использую эту линию. - person Kes; 26.02.2017