You can just pass them to the initialization of your Datasets:
train_dataset = ImageFolder(
    data_dir,
    transform=train_transform,
)

val_dataset = ImageFolder(
    data_dir,
    transform=validation_transform
)