pytorch-lightning - 在 pytorch 闪电中组织数据转换的最佳方法

Question

我正在训练具有各种图像尺寸（50x50 到 200x200）的模型，我想尝试不同的模型，例如 torchvision 中可用的预训练模型。

转换示例：

transforms.Compose([
            transforms.ToPILImage(),
            transforms.Resize(256),
            transforms.CenterCrop(224),
            transforms.ToTensor(),
            transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
        ])

放置此代码的最佳位置是什么，因此如果我使用不同的模型，我可以轻松更改其参数（例如：如果我需要更改输入图像大小/标准化参数？我还在使用 imgaug 库进行训练过程中的一些图像增强。