训练数据前置处理与提升训练效率

configs/test.yml

@@ -8,13 +8,13 @@ base:
   log_level: "info"                   # 日志级别（debug/info/warning/error）
   embedding_size: 256                 # 特征维度
   pin_memory: true                    # 是否启用pin_memory
-  distributed: true                  # 是否启用分布式训练
+  distributed: false                  # 是否启用分布式训练
 
 # 模型配置
 models:
   backbone: 'resnet18'
   channel_ratio: 1.0
-  model_path: "checkpoints/resnet18_scatter_6.26/best.pth"
+  model_path: "checkpoints/resnet18_scatter_7.3/best.pth"
   half: false                        # 是否启用半精度测试（fp16）
   contrast_learning: false
 
@@ -22,9 +22,9 @@ models:
 data:
   test_batch_size: 128               # 训练批次大小
   num_workers: 32                      # 数据加载线程数
-  test_dir: "../data_center/scatter/v2/val_extar"   # 验证数据集根目录
+  test_dir: "../data_center/scatter/v4/val"   # 验证数据集根目录
   test_group_json: "../data_center/contrast_learning/model_test_data/test/inner_group_pairs.json"
-  test_list: "../data_center/scatter/val_extar_cross_same.txt"
+  test_list: "../data_center/scatter/v4/standard_cross_same.txt"
   group_test: false
   save_image_joint: true
   image_joint_pth: "./joint_images"
@@ -37,6 +37,11 @@ transform:
   RandomRotation: 180               # 随机旋转角度
   ColorJitter: 0.5                   # 随机颜色抖动强度
 
+heatmap:
+    image_joint_pth: "./heatmap_joint_images"
+    feature_layer: "layer4"
+    show_heatmap: true
+
 save:
   save_dir: ""
   save_name: ""