# 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" image_joint_pth: "/home/lc/data_center/image_analysis/error_compare_result" total_pkl: "/home/lc/data_center/image_analysis/pic_pic_similar_maxtrix/total.pkl" result_txt: "/home/lc/data_center/image_analysis/pic_pic_similar_maxtrix/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"