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: 是否进行随机擦除。
Github 上地址 github.com/thuml/TimesNet. 的开源项目 TIMESNET 在 `data_loader.py` 中 class Dataset_Custom(Dataset) 每个输入参数的具体含义

原文地址: https://www.cveoy.top/t/topic/yIG 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录