如何解决在 pytorch 闪电中组织数据转换的最佳方式
我正在训练具有各种图像尺寸(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 库。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。