feat : 代码梳理 移除所有敏感密钥 通过环境变量方式配置

app/service/clothing_seg/service.py

@@ -9,14 +9,14 @@ from PIL import Image
 from minio import Minio
 from tritonclient.utils import np_to_triton_dtype
 
-from app.core.config import *
+from app.core.config import settings
 from app.schemas.clothing_seg import ClothingSegModel
 from app.service.design_fast.utils.design_ensemble import get_seg_result
 from app.service.utils.decorator import RunTime
 from app.service.utils.generate_uuid import generate_uuid
 from app.service.utils.new_oss_client import oss_get_image, oss_upload_image
 
-minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
+minio_client = Minio(settings.MINIO_URL, access_key=settings.MINIO_ACCESS, secret_key=settings.MINIO_SECRET, secure=settings.MINIO_SECURE)
 
 
 class ClothingSeg:
@@ -64,9 +64,9 @@ class ClothingSeg:
             if image_type == "sketch":
                 if len(image.shape) == 2:
                     image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
-                    seg_mask = get_seg_result(1, image[:, :, :3])
+                    seg_mask = get_seg_result(image[:, :, :3])
                 else:
-                    seg_mask = get_seg_result(1, image[:, :, :3])
+                    seg_mask = get_seg_result(image[:, :, :3])
                 temp = seg_mask != 0.0
                 mask = (255 * (temp + 0).astype(np.uint8))
                 x_min, y_min, x_max, y_max = get_bounding_box(mask)