Преобразование изображения в TensorFloat в Windows ML (для преобразованной модели Keras)

Question

Я следовал этому официальному учебнику, чтобы создать приложение UWP, которое может определять рукописные цифры с использованием предварительно обученной модели.

Но мне нужно использовать модели Keras.

Поэтому я преобразовал эту модель Keras из примеров Keras, которая также обучается на наборе данных MNIST с winmltools в python в ONNX:

onnx_model = winmltools.convert_keras(model, 7, name='mnist')   
print(onnx_model.graph.input)   
winmltools.save_model(onnx_model, "mnist_tf.onnx")`

Когда я импортирую ее в Visual Studio mlgen генерирует этот код для ввода:

public TensorFloat conv2d_1_input; // shape(-1,28,28,1)

Но на оригинальной модели из учебника это:

public ImageFeatureValue Input3; // BitmapPixelFormat: Gray8, BitmapAlphaMode: Premultiplied, width: 28, height: 28

Как можно Я конвертирую ImageFeatureValue в TensorFloat?

Мне также нужно импортировать файлы изображений, но я нашел только способы, которые также использовали ImageFeatureValue, а не TensorFloat.

Спасибо!

Answer 1

Я бы просто изменил код, сгенерированный инструментом mlgen в Visual Studio, чтобы он соответствовал коду учебного примера! Я иногда обнаруживал, что инструмент mlgen выводит какой-то вонкий код для форматирования модели ...