kt.py

import os

def count_images_in_folders(root_dir):
    class_counts = {}
    for class_name in os.listdir(root_dir):
        class_path = os.path.join(root_dir, class_name)
        if os.path.isdir(class_path):
            num_images = len([f for f in os.listdir(class_path) if f.lower().endswith(('.png', '.jpg', '.jpeg'))])
            class_counts[class_name] = num_images
    return class_counts

# Kiểm tra số lượng ảnh trong tập train và val
train_counts = count_images_in_folders("val")
val_counts = count_images_in_folders("train")
test_counts = count_images_in_folders("test")
print("Số lượng ảnh val:", train_counts)
print("Số lượng ảnh train:", val_counts)
print("Số lượng ảnh test:", test_counts)