|
|
@@ -106,7 +106,7 @@ def watermark_embed():
|
|
|
classification_pytorch_black_embed.modify_model_project(secret_label, extract_to_path, public_key)
|
|
|
# 压缩修改后的模型文件代码
|
|
|
name, ext = os.path.splitext(file_name)
|
|
|
- zip_filename = f"{name}_{embed_type}_embed{ext}"
|
|
|
+ zip_filename = f"{model_value}_{'tensorflow' if 'tensorflow' in model_file else 'pytorch'}_{embed_type}_embed{ext}"
|
|
|
zip_filepath = os.path.join(file_path, zip_filename)
|
|
|
logger.info(f"zip modified model project source to {zip_filepath}")
|
|
|
with zipfile.ZipFile(zip_filepath, 'w', zipfile.ZIP_DEFLATED) as zipf:
|
