yolo_annot_count.py

import glob, os 
import sys

def yolo_reader(annotation_file):

    with open(annotation_file, 'r') as f:
        data = f.read()
    data = data.split("\n")
    return data

# <class id> <Xo/X> <Yo/Y> <W/X> <H/Y>
def collate_classes(annotation_files):
    classes = {}
    for f in annotation_files:
        data = yolo_reader(f)
        for i in data:
            cls = i.split(" ")[0]
            classes[cls] = classes.get(cls, 0) +1
    return classes

def main(dir):
    os.chdir(dir)
    annotation_files = []
    for file in glob.glob("*.txt"):
        annotation_files.append(file)

    classes = collate_classes(annotation_files)
    print(classes)


if __name__ == '__main__':
    if len(sys.argv) >1:
        main(sys.argv[1].rstrip())