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

更改

Browse Source
This commit is contained in:
lee
2025-06-18 11:30:55 +08:00
parent 180a41ae90
commit bf9604ec29
6 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
.idea/CopilotChatHistory.xml generated
View File

@ -3,6 +3,18 @@
<component name="CopilotChatHistory"> <component name="CopilotChatHistory">
<option name="conversations"> <option name="conversations">
<list> <list>
<Conversation>
<option name="createTime" value="1749793513436" />
<option name="id" value="019767d21fdc756ba782b33c8b14cdf1" />
<option name="title" value="新对话 2025年6月13日 13:45:13" />
<option name="updateTime" value="1749793513436" />
</Conversation>
<Conversation>
<option name="createTime" value="1749792408202" />
<option name="id" value="019767c1428a7aa28682039d57d19778" />
<option name="title" value="新对话 2025年6月13日 13:26:48" />
<option name="updateTime" value="1749792408202" />
</Conversation>
<Conversation> <Conversation>
<option name="createTime" value="1749718122230" /> <option name="createTime" value="1749718122230" />
<option name="id" value="01976353bef6703884544447c919013c" /> <option name="id" value="01976353bef6703884544447c919013c" />

2
configs/compare.yml
View File

@ -62,7 +62,7 @@ transform:
# 日志与监控 # 日志与监控
logging: logging:
logging_dir: "./logs" # 日志保存目录 logging_dir: "./logs/resnet50_log" # 日志保存目录
tensorboard: true # 是否启用TensorBoard tensorboard: true # 是否启用TensorBoard
checkpoint_interval: 30 # 检查点保存间隔（epoch） checkpoint_interval: 30 # 检查点保存间隔（epoch）

2
configs/distill.yml
View File

@ -51,7 +51,7 @@ data:
dataset: "imagenet" # 数据集名称（示例用，可替换为实际数据集） dataset: "imagenet" # 数据集名称（示例用，可替换为实际数据集）
train_batch_size: 128 # 训练批次大小 train_batch_size: 128 # 训练批次大小
val_batch_size: 100 # 验证批次大小 val_batch_size: 100 # 验证批次大小
num_workers: 4 # 数据加载线程数 num_workers: 16 # 数据加载线程数
data_train_dir: "../data_center/contrast_learning/data_base/train" # 训练数据集根目录 data_train_dir: "../data_center/contrast_learning/data_base/train" # 训练数据集根目录
data_val_dir: "../data_center/contrast_learning/data_base/val" # 验证数据集根目录 data_val_dir: "../data_center/contrast_learning/data_base/val" # 验证数据集根目录

1
configs/test.yml
View File

@ -19,7 +19,6 @@ models:
# 数据配置 # 数据配置
data: data:
group_test: False # 数据集名称（示例用，可替换为实际数据集）
test_batch_size: 128 # 训练批次大小 test_batch_size: 128 # 训练批次大小
num_workers: 32 # 数据加载线程数 num_workers: 32 # 数据加载线程数
test_dir: "../data_center/contrast_learning/contrast_test_data" # 验证数据集根目录 test_dir: "../data_center/contrast_learning/contrast_test_data" # 验证数据集根目录

5
configs/write_feature.yml
View File

@ -22,7 +22,7 @@ data:
test_batch_size: 128 # 验证批次大小 test_batch_size: 128 # 验证批次大小
num_workers: 32 # 数据加载线程数 num_workers: 32 # 数据加载线程数
half: true # 是否启用半精度数据 half: true # 是否启用半精度数据
img_dirs_path: "/shareData/temp_data/comparison/Hangzhou_Yunhe/base_data/05-09" img_dirs_path: "/personalDocument/lic/contrast_base"
# img_dirs_path: "/home/lc/contrast_nettest/data/feature_json" # img_dirs_path: "/home/lc/contrast_nettest/data/feature_json"
xlsx_pth: false # 过滤商品， 默认None不进行过滤 xlsx_pth: false # 过滤商品， 默认None不进行过滤
@ -42,6 +42,7 @@ logging:
save: save:
json_bin: "../search_library/yunhedian_05-09.json" # 保存整个json文件 json_bin: "../search_library/yunhedian_05-09.json" # 保存整个json文件
json_path: "../data/feature_json_compare/" # 保存单个json文件 json_path: "../feature_json/" # 保存单个json文件
error_barcodes: "error_barcodes.txt" error_barcodes: "error_barcodes.txt"
barcodes_statistics: "../search_library/barcodes_statistics.txt" barcodes_statistics: "../search_library/barcodes_statistics.txt"
create_single_json: true

2
tools/write_feature_json.py
View File

@ -407,5 +407,5 @@ if __name__ == "__main__":
column_values = extractor.get_shop_barcodes(conf['data']['xlsx_pth']) column_values = extractor.get_shop_barcodes(conf['data']['xlsx_pth'])
imgs_dict = extractor.get_files(conf['data']['img_dirs_path'], imgs_dict = extractor.get_files(conf['data']['img_dirs_path'],
filter=column_values, filter=column_values,
create_single_json=False) # False create_single_json=conf['save']['create_single_json']) # False
extractor.statisticsBarcodes(conf['data']['img_dirs_path'], column_values) extractor.statisticsBarcodes(conf['data']['img_dirs_path'], column_values)