def count_images_in_classes(lbls):
    """
    Determining number of images in a class for Graph
    :param lbls: Labels (different classes)
    :return: Dictionary of labels and count
    """
    dct = {}
    for i in lbls:
        if i in dct:
            dct[i] += 1
        else:
            dct[i] = 1
    return dct