feat(actpred): 训练时安全图像增强(ColorJitter),支持任意 frame_len

训练集启用 --augment 后对每帧施加亮度/对比度/饱和度/色调变换,
同一样本内所有帧共享参数(模拟连续帧光照不变),验证集不增强。
仅颜色空间变换,无空间变换,动作标签不受影响。
删除 t1.mt(@TEST 块无对应插件,录制时报错)。
This commit is contained in:
qing committed 2026-08-13 09:35:37 +00:00
1 parent 85133868f3
commit 5af863a2c5
5 files changed
+136 -23

No files matched your search

+2 -1
View File
@@ -84,7 +84,8 @@ def main():
train_ds = ActionDataset(
cfg.data.data_dir, cfg.data.history_len, cfg.data.action_dim,
cfg.data.image_width, cfg.data.image_height, cfg.data.normalize_actions,
cfg.data.action_stats_path, train_indices, cfg.data.frame_len)
cfg.data.action_stats_path, train_indices, cfg.data.frame_len,
augment=cfg.train.augment)
if cfg.data.normalize_actions and not cfg.data.action_stats_path:
# 归一化统计量必须只用训练集计算,再供验证集复用,否则 val loss 不可信
stats_path = os.path.join(cfg.train.checkpoint_dir, "action_stats.npy")