修改faster-rcnn白盒水印嵌入方式

watermark_generate/deals/faster_rcnn_pytorch_white_embed.py

@@ -19,7 +19,7 @@ def modify_model_project(secret_label: str, project_dir: str, public_key: str):
     project_dir = os.path.join(project_dir, rela_project_path[0])
     project_file = os.path.join(project_dir, 'utils/utils_fit.py')
     project_file2 = os.path.join(project_dir, 'nets/frcnn_training.py')
-    project_file3 = os.path.join(project_dir, 'nets/train.py')
+    project_file3 = os.path.join(project_dir, 'train.py')
 
     if not os.path.exists(project_file) or not os.path.exists(project_file2):
         raise BusinessException(message="指定待修改的工程文件未找到", code=-1)