File size: 1,311 Bytes
bd421ea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import argparse
import numpy as np

from dataset import read_IAM_label_txt_file

def list_unique_characters_in_IAM_dataset(FLAGS):
    _, all_labels = read_IAM_label_txt_file(FLAGS.file_txt_labels)

    num_labels = len(all_labels)
    print(f"num labels : {num_labels}")
    unique_chars = []

    for label in all_labels:
        unique_chars = unique_chars + list(np.unique(np.array(list(label))))

    unique_chars = sorted(unique_chars)
    unique_chars = np.array(unique_chars)
    unique_chars = np.unique(unique_chars)
    unique_chars = ''.join(unique_chars)
    
    # prints all unique chars in the IAM dataset
    print(unique_chars)

    # prints the number of unique chars in the IAM dataset
    print(f"Number of unique characters : {len(unique_chars)}")
    return

def main():
    file_txt_labels = "/home/abhishek/Desktop/RUG/hw_recognition/IAM-data/iam_lines_gt.txt"

    parser = argparse.ArgumentParser(
        formatter_class=argparse.ArgumentDefaultsHelpFormatter
    )

    parser.add_argument("--file_txt_labels", default=file_txt_labels,
        type=str, help="full path to label text file")

    FLAGS, unparsed = parser.parse_known_args()
    list_unique_characters_in_IAM_dataset(FLAGS)
    return

if __name__ == "__main__":
    main()