屏蔽/255,避免重复除以。

This commit is contained in:
zhanghuan
2025-09-11 17:49:37 +08:00
parent 06e440a6d1
commit e1cb936e32
4 changed files with 10 additions and 6 deletions
+2 -2
View File
@@ -8,7 +8,7 @@ def main():
# 1. 加载训练好的基础模型权重
base_model = create_food_cnn()
model_path = "../model/07/best_food_model.pth"
model_path = "../model/09/best_food_model.pth"
if not os.path.exists(model_path):
print(f"错误:模型文件不存在 {model_path}")
@@ -37,7 +37,7 @@ def main():
traced_model = torch.jit.trace(mobile_model, example_input)
# 保存模型
output_path = "../model/07/best_food_model_mobile.pt"
output_path = "../model/09/best_food_model_mobile.pt"
traced_model.save(output_path)
print(f"✓ TorchScript模型保存成功: {output_path}")