feat generate 污点图保存

2024-05-17 18:22:00 +08:00
parent bb46282987
commit 18eb4735ea
1 changed files with 1 additions and 0 deletions
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -80,6 +80,7 @@ GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg"
 # SEG service config
 SEG_MODEL_URL = '10.1.1.240:10000'
 SEGMENTATION = {
+    "new_model_name": "seg_knet",
    "name": "seg_ocrnet_hr18",
    "input": "seg_input__0",
    "output": "seg_output__0",