lichen/ieemoo-ai-contrast
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

更改

Browse Source
This commit is contained in:
lee
2025-07-02 14:41:12 +08:00
parent 061820c34f
commit 537ed838fc
8 changed files with 124 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
configs/scatter.yml
View File

@ -18,20 +18,20 @@ models:
# 训练参数
training:
epochs: 300 # 总训练轮次
epochs: 600 # 总训练轮次
batch_size: 64 # 批次大小
lr: 0.005 # 初始学习率
lr: 0.0004 # 初始学习率
optimizer: "sgd" # 优化器类型
metric: 'arcface' # 损失函数类型（可选：arcface/cosface/sphereface/softmax）
loss: "cross_entropy" # 损失函数类型（可选：cross_entropy/cross_entropy_smooth/center_loss/center_loss_smooth/arcface/cosface/sphereface/softmax）
lr_step: 10 # 学习率调整间隔（epoch）
lr_decay: 0.98 # 学习率衰减率
lr_decay: 0.95 # 学习率衰减率
weight_decay: 0.0005 # 权重衰减
scheduler: "cosine_annealing" # 学习率调度器（可选：cosine_annealing/step/none）
scheduler: "step" # 学习率调度器（可选：cosine_annealing/step/none）
num_workers: 32 # 数据加载线程数
checkpoints: "./checkpoints/resnet18_scatter_6.2/" # 模型保存目录
checkpoints: "./checkpoints/resnet18_scatter_6.26/" # 模型保存目录
restore: True
restore_model: "checkpoints/resnet18_scatter_6.2/best.pth" # 模型恢复路径
restore_model: "checkpoints/resnet18_scatter_6.25/best.pth" # 模型恢复路径
@ -46,8 +46,8 @@ data:
train_batch_size: 128 # 训练批次大小
val_batch_size: 100 # 验证批次大小
num_workers: 32 # 数据加载线程数
data_train_dir: "../data_center/scatter/train" # 训练数据集根目录
data_val_dir: "../data_center/scatter/val" # 验证数据集根目录
data_train_dir: "../data_center/scatter/v2/train" # 训练数据集根目录
data_val_dir: "../data_center/scatter/v2/val" # 验证数据集根目录
transform:
img_size: 224 # 图像尺寸
@ -59,7 +59,7 @@ transform:
# 日志与监控
logging:
logging_dir: "./log/2025.6.2-scatter.txt" # 日志保存目录
logging_dir: "./log/2025.6.25-scatter.txt" # 日志保存目录
tensorboard: true # 是否启用TensorBoard
checkpoint_interval: 30 # 检查点保存间隔（epoch）

12
configs/test.yml
View File

@ -13,17 +13,21 @@ base:
# 模型配置
models:
backbone: 'resnet18'
channel_ratio: 0.75
model_path: "./checkpoints/resnet18_1009/best.pth"
channel_ratio: 1.0
model_path: "checkpoints/resnet18_scatter_6.26/best.pth"
half: false # 是否启用半精度测试（fp16）
contrast_learning: false
# 数据配置
data:
test_batch_size: 128 # 训练批次大小
num_workers: 32 # 数据加载线程数
test_dir: "../data_center/contrast_learning/contrast_test_data" # 验证数据集根目录
test_dir: "../data_center/scatter/v2/val_extar" # 验证数据集根目录
test_group_json: "../data_center/contrast_learning/model_test_data/test/inner_group_pairs.json"
test_list: "../data_center/contrast_learning/contrast_test_data/cross_same.txt"
test_list: "../data_center/scatter/val_extar_cross_same.txt"
group_test: false
save_image_joint: true
image_joint_pth: "./joint_images"
transform:
img_size: 224 # 图像尺寸

4
configs/transform.yml
View File

@ -18,8 +18,8 @@ models:
channel_ratio: 0.75
model_path: "../checkpoints/resnet18_1009/best.pth"
onnx_model: "../checkpoints/resnet18_1009/best.onnx"
rknn_model: "../checkpoints/resnet18_1009/best_rknn2.3.2.rknn"
rknn_batch_size: 1
rknn_model: "../checkpoints/resnet18_1009/best_rknn2.3.2_batch16.rknn"
rknn_batch_size: 16
# 日志与监控
logging: