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]
def __init__(self, blank_index): self.idx2char_dict = Config.get_idx2char() self.blank_index = blank_index
def get_num_classes_and_blank_index(): num_classes = len(Config.get_idx2char()) blank_index = num_classes - 1 return num_classes, blank_index