Exemplo n.º 1
0
def get_final_output_string(output, blank_index):
    decoded_text = tf.cast(x=output, dtype=tf.int32)
    decoded_text = decoded_text.numpy()
    decoded_text = index_to_char(inputs=decoded_text, idx2char_dict=Config.get_idx2char(), blank_index=blank_index)
    return decoded_text[0]
Exemplo n.º 2
0
 def __init__(self, blank_index):
     self.idx2char_dict = Config.get_idx2char()
     self.blank_index = blank_index
Exemplo n.º 3
0
def get_num_classes_and_blank_index():
    num_classes = len(Config.get_idx2char())
    blank_index = num_classes - 1
    return num_classes, blank_index