|
@@ -81,7 +81,6 @@ def watermark_embed():
|
|
|
|
|
|
|
|
|
logger.info(f"modify model project source, model_value: {model_value}, embed_type: {embed_type}")
|
|
|
-
|
|
|
if "tensorflow" in model_file:
|
|
|
if (model_value in ['alexnet', 'vggnet']) and embed_type == 'whitebox':
|
|
|
classfication_tensorflow_white_embed.modify_model_project(secret_label, extract_to_path, public_key)
|