Browse Source

删除多余参数

zhy 2 weeks ago
parent
commit
a21cbdf5a7
1 changed files with 3 additions and 3 deletions
  1. 3 3
      watermark_verify/app.py

+ 3 - 3
watermark_verify/app.py

@@ -17,9 +17,9 @@ def create_app():
         # TODO 根据工标需要进行HTTP接口开发
         data = request.json
         model_filename = data.get('model_filename') # 模型权重文件位置
-        framework = data.get('framework', 'pytorch')
-        mode = data.get('mode', 'blackbox')
-        model_type = data.get('model_type', 'ssd')
+        framework = data.get('framework')
+        mode = data.get('mode')
+        model_type = data.get('model_type')
 
         result = verify_tool.label_verification(model_filename=model_filename, framework=framework, mode=mode, model_type=model_type)
         print(f"模型水印检测结果: {result}")