CTCBeamSearchDecoder считает, что sequence_length of shape (2,) не является вектором

Пытаясь запустить поиск луча в модели Keras, я получаю запутанные (и противоречивые?) Сообщения об ошибках. Моя модель имеет такие входные данные, как

inputs = Input(name='spectrograms',
               shape=(None, hparams["n_spectrogram"]))
input_length = Input(name='len_spectrograms',
                     shape=[1], dtype='int64')

и функция потери CTC требует [1] формы на входе и длина метки. Насколько я понимаю, вывод должен быть получен с чем-то вроде

# Stick connectionist temporal classification on the end of the core model
paths = K.function(
    [inputs, input_length],
    K.ctc_decode(output, input_length, greedy=False, top_paths=4)[0])

но как есть, это приводит к жалобе на форму input_length

ValueError: Shape must be rank 1 but is rank 2 for 'CTCBeamSearchDecoder' (op: 'CTCBeamSearchDecoder') with input shapes: [?,?,44], [?,1].

но если я отрублю это измерение

    K.ctc_decode(output, input_length[..., 0], greedy=False, top_paths=4)[0])

определение модели выполняется, но когда я запускаю y = paths([x, numpy.array([[30], [30]])]) с x.shape == (2, 30, 513) Я вдруг получаю

tensorflow.python.framework.errors_impl.InvalidArgumentError: sequence_length is not a vector
     [[{{node CTCBeamSearchDecoder}} = CTCBeamSearchDecoder[beam_width=100, merge_repeated=true, top_paths=4, _device="/job:localhost/replica:0/task:0/device:CPU:0"](Log, ToInt32)]]

Что я делаю неправильно?

0 ответов

Другие вопросы по тегам