training/test_loader.py

#Testing the dataloader functionality
from dataloader import get_dataloaders

train_loader, val_loader = get_dataloaders(
    csv_path="data_processed/metadata_final.csv",
    images_dir="data_processed/images",
    batch_size=32
)

images, labels = next(iter(train_loader))
print("Batch image shape:", images.shape)
print("Batch labels shape:", labels.shape)