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-10 14:24:05 +08:00
parent 0701538a73
commit 09f41f6289
15 changed files with 430 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
configs/transform.yml
View File

@ -15,11 +15,11 @@ base:
# 模型配置
models:
backbone: 'resnet18'
channel_ratio: 0.75
channel_ratio: 1.0
model_path: "../checkpoints/resnet18_1009/best.pth"
onnx_model: "../checkpoints/resnet18_1009/best.onnx"
rknn_model: "../checkpoints/resnet18_1009/best_rknn2.3.2_batch16.rknn"
rknn_batch_size: 16
onnx_model: "../checkpoints/resnet18_3399_sancheng/best.onnx"
rknn_model: "../checkpoints/resnet18_3399_sancheng/best_rknn2.3.2_RK3566.rknn"
rknn_batch_size: 1
# 日志与监控
logging: