TIMESNET 数据集加载器参数详解 - Github.com/thuml/TimesNet

TIMESNET 数据集加载器参数详解 - Github.com/thuml/TimesNet

本文将详细介绍 Github.com/thuml/TimesNet 开源项目中 data_loader.py 文件中 class Dataset_Custom(Dataset) 的每个输入参数的含义。

参数解释:

1. 'data_path': 数据集的路径。
2. 'label_path': 标签集的路径。
3. 'crop_size': 图像的裁剪尺寸。
4. 'is_train': 是否是训练模式。
5. 'mean': 数据集的均值。
6. 'std': 数据集的标准偏差。
7. 'scale': 数据集的缩放尺度。
8. 'flip': 是否进行图像翻转。
9. 'rotate': 是否进行图像旋转。
10. 'color_jitter': 是否进行图像颜色增强。
11. 'normalize': 是否对图像进行标准化。
12. 'to_tensor': 是否将图像转换为张量。
13. 'random_erase': 是否进行随机擦除。