55 lines
2.2 KiB
YAML
55 lines
2.2 KiB
YAML
# configs/similar_analysis.yml
|
||
# 专为模型训练对比设计的配置文件
|
||
# 支持对比不同训练策略（如蒸馏vs独立训练）
|
||
|
||
# 基础配置
|
||
base:
|
||
experiment_name: "model_comparison" # 实验名称（用于结果保存目录）
|
||
device: "cuda" # 训练设备（cuda/cpu）
|
||
embedding_size: 256 # 特征维度
|
||
pin_memory: true # 是否启用pin_memory
|
||
distributed: true # 是否启用分布式训练
|
||
|
||
|
||
# 模型配置
|
||
models:
|
||
backbone: 'resnet18'
|
||
channel_ratio: 0.75
|
||
model_path: "../checkpoints/resnet18_20250715_scale=0.75_sub/best.pth"
|
||
# model_path: "../checkpoints/resnet18_1009/best.pth"
|
||
|
||
heatmap:
|
||
feature_layer: "layer4"
|
||
show_heatmap: true
|
||
# 数据配置
|
||
data:
|
||
dataset: "imagenet" # 数据集名称（示例用，可替换为实际数据集）
|
||
train_batch_size: 128 # 训练批次大小
|
||
val_batch_size: 8 # 验证批次大小
|
||
num_workers: 32 # 数据加载线程数
|
||
data_dir: "/home/lc/data_center/image_analysis/pic_pic_similar_maxtrix_new"
|
||
image_joint_pth: "/home/lc/data_center/image_analysis/error_compare_result"
|
||
total_pkl: "/home/lc/data_center/image_analysis/pic_pic_similar_maxtrix_new/total.pkl"
|
||
result_txt: "/home/lc/data_center/image_analysis/pic_pic_similar_maxtrix_new/result.txt"
|
||
|
||
transform:
|
||
img_size: 224 # 图像尺寸
|
||
img_mean: 0.5 # 图像均值
|
||
img_std: 0.5 # 图像方差
|
||
RandomHorizontalFlip: 0.5 # 随机水平翻转概率
|
||
RandomRotation: 180 # 随机旋转角度
|
||
ColorJitter: 0.5 # 随机颜色抖动强度
|
||
|
||
# 日志与监控
|
||
logging:
|
||
logging_dir: "./logs/resnet18_scale=0.75_nosub_log" # 日志保存目录
|
||
tensorboard: true # 是否启用TensorBoard
|
||
checkpoint_interval: 30 # 检查点保存间隔（epoch）
|
||
|
||
#event:
|
||
# oneToOne_max_th: 0.9
|
||
# oneToSn_min_th: 0.6
|
||
# event_save_dir: "/home/lc/works/realtime_yolov10s/online_yolov10s_resnetv11_20250702/yolos_tracking"
|
||
# stdlib_image_path: "/testDataAndLogs/module_test_record/comparison/标准图测试数据/pic/stlib_base"
|
||
# pickle_path: "event.pickle"
|