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