Как получить изображения имен файлов из мини-бара?

Question

Я работаю на этом уроке:

https://github.com/Microsoft/CNTK/blob/master/Tutorials/CNTK_201B_CIFAR-10_ImageHandsOn.ipynb

Тест/файлы данных поезда простые разделенные табуляцией текстовые файлы, содержащие имена файлов изображений и правильные ярлыки, как это:

...\data\CIFAR-10\test\00000.png 3 
...\data\CIFAR-10\test\00001.png 8 
...\data\CIFAR-10\test\00002.png 8 

Предположим я создаю такой как:

test_minibatch = reader_test.next_minibatch(10) 

Как я могу получить имена файлов для изображений, которые w как в первом столбце файла тестовых данных?

Я пытался с этим кодом:

orig_features = np.asarray(test_minibatch[features_stream_info].m_data) 
print(orig_features) 

Но, что приводит к печати байт самих изображений.

Answer 1

Имя файла теряется при загрузке изображений через устройство чтения изображений.

Одним из возможных решений является использование композитного считывателя для одновременного загрузки файла карты в текстовом формате. У нас есть композитный пример читателя здесь с BrainScript: https://github.com/Microsoft/CNTK/tree/master/Examples/Image/Regression

С Python, вы могли бы сделать что-то вроде:

# read images 
image_source = ImageDeserializer(map_file) 
image_source.ignore_labels() 
image_source.map_features(features_stream_name, 
    [ImageDeserializer.scale(width=image_width, height=image_height, channels=num_channels, 
          scale_mode="pad", pad_value=114, interpolations='linear')]) 

# read rois and labels 
roi_source = CTFDeserializer(roi_file) 
roi_source.map_input(rois_stream_name, dim=rois_dim, format="dense") 
label_source = CTFDeserializer(label_file) 
label_source.map_input(labels_stream_name, dim=label_dim, format="dense") 

# define a composite reader 
rc = ReaderConfig([image_source, roi_source, label_source], epoch_size=sys.maxsize) 
return rc.minibatch_source()