TensorRT 5 UFFParser ошибка весов ядра - PullRequest
Новая
       35

TensorRT 5 UFFParser ошибка весов ядра

0 голосов
/ 02 мая 2019

Я пытаюсь преобразовать, затем выполнить вывод с помощью модели Tensorflow.

Я создаю модель Tensorflow с помощью tf.keras: 

model = Sequential()
model.add(Conv2D(64, kernel_size=3, activation='relu', input_shape=(224,224,3), name='image'))
model.add(Conv2D(32, kernel_size=3, activation='relu'))
model.add(Flatten())
model.add(Dense(num_classes, activation='softmax', name='output'))

Я обучаю модель, а затем преобразовываюфайл, сохраняющий замороженный графовый файл тензорного потока .pb, преобразованный в UFF с: 

uff_model = uff.from_tensorflow_frozen_model(
frozen_file="model/tensorflow/simplemodel.pb",
output_nodes=["output/Softmax"],
output_filename="model/tensorrt/simplemodel.uff")

Когда я анализирую модель с: 

TRT_LOGGER = trt.Logger(trt.Logger.INFO)
builder = trt.Builder(TRT_LOGGER)
network = builder.create_network()
parser = trt.UffParser()
parser.register_input("image_input", (1, 224, 224, 3))
parser.register_output("output/SoftMax")
parser.parse("model/tensorrt/simplemodel.uff", network)

, я получил эту ошибку: 

[TensorRT] INFO: UFFParser: parsing image_input
[TensorRT] INFO: UFFParser: parsing image/kernel
[TensorRT] INFO: UFFParser: parsing image/Conv2D
[TensorRT] INFO: UFFParser: parsing image/bias
[TensorRT] INFO: UFFParser: parsing image/BiasAdd
[TensorRT] ERROR: image/Conv2D: kernel weights has count 1728 but 129024 was expected
[TensorRT] ERROR: UFFParser: Parser error: image/BiasAdd: The input to the Scale Layer is required to have a minimum of 3 dimensions.

Почему ошибка подсчета веса ядра считается?Я установил входную форму с правильным размером (224 244,3 ...), это моя модель: 

Layer (type)                 Output Shape              Param #
=================================================================
image (Conv2D)               (None, 222, 222, 64)      1792
_________________________________________________________________
conv2d (Conv2D)              (None, 220, 220, 32)      18464
_________________________________________________________________
flatten (Flatten)            (None, 1548800)           0
_________________________________________________________________
output (Dense)               (None, 6)                 9292806
=================================================================
Total params: 9,313,062
Trainable params: 9,313,062
Non-trainable params: 0

Спасибо,

Игорь

...