84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
简单的模型解析调试
|
||
"""
|
||
|
||
import os
|
||
import glob
|
||
|
||
def debug_model_files():
|
||
"""调试模型文件"""
|
||
|
||
# 获取模型目录
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
model_dir = os.path.join(current_dir, 'saved_models')
|
||
|
||
print(f"模型目录: {model_dir}")
|
||
print(f"目录存在: {os.path.exists(model_dir)}")
|
||
|
||
if not os.path.exists(model_dir):
|
||
print("模型目录不存在!")
|
||
return
|
||
|
||
# 列出所有.pth文件
|
||
pattern = os.path.join(model_dir, "*.pth")
|
||
model_files = glob.glob(pattern)
|
||
|
||
print(f"找到 {len(model_files)} 个模型文件:")
|
||
for model_file in model_files:
|
||
filename = os.path.basename(model_file)
|
||
print(f" {filename}")
|
||
|
||
# 测试解析逻辑
|
||
model_id = filename.replace('.pth', '')
|
||
print(f" 生成的model_id: {model_id}")
|
||
|
||
# 测试文件名解析（复制ModelManager的逻辑）
|
||
model_info = parse_model_filename(filename)
|
||
print(f" 解析结果: {model_info}")
|
||
print()
|
||
|
||
def parse_model_filename(filename):
|
||
"""
|
||
简化的文件名解析逻辑（复制自ModelManager）
|
||
"""
|
||
if not filename.endswith('.pth'):
|
||
return None
|
||
|
||
base_name = filename.replace('.pth', '')
|
||
|
||
try:
|
||
# 新格式解析
|
||
if '_product_' in base_name:
|
||
# 产品模式: model_type_product_product_id_version
|
||
parts = base_name.split('_product_')
|
||
model_type = parts[0]
|
||
rest = parts[1]
|
||
|
||
# 分离产品ID和版本
|
||
if '_v' in rest:
|
||
last_v_index = rest.rfind('_v')
|
||
product_id = rest[:last_v_index]
|
||
version = rest[last_v_index+1:]
|
||
else:
|
||
product_id = rest
|
||
version = 'v1'
|
||
|
||
return {
|
||
'model_type': model_type,
|
||
'product_id': product_id,
|
||
'version': version,
|
||
'training_mode': 'product',
|
||
'store_id': None,
|
||
'aggregation_method': None
|
||
}
|
||
|
||
# 如果不匹配任何格式
|
||
return None
|
||
|
||
except Exception as e:
|
||
print(f"解析失败: {e}")
|
||
return None
|
||
|
||
if __name__ == "__main__":
|
||
debug_model_files() |