|
@@ -81,7 +81,6 @@ def watermark_embed():
|
|
|
|
|
|
# 修改模型文件代码,并将public_key写入至文件保存至修改后的工程文件目录中
|
|
|
logger.info(f"modify model project source, model_value: {model_value}, embed_type: {embed_type}")
|
|
|
- # TODO 添加其他模型工程代码处理
|
|
|
if "tensorflow" in model_file: # tensorflow、keras框架水印嵌入支持
|
|
|
if (model_value in ['alexnet', 'vggnet']) and embed_type == 'whitebox':
|
|
|
classfication_tensorflow_white_embed.modify_model_project(secret_label, extract_to_path, public_key)
|