feat:

fix:  替换项目中所有mmcv的依赖

app/service/lineart/service.py

@@ -1,7 +1,6 @@
 import logging
 
 import cv2
-import mmcv
 import numpy as np
 import torch
 import torch.nn.functional as F
@@ -10,6 +9,7 @@ from minio import Minio
 from app.core.config import settings
 from app.core.config import DESIGN_MODEL_URL
 from app.schemas.image2sketch import Image2SketchModel
+from app.service.utils.image_normalize import my_imnormalize
 from app.service.utils.new_oss_client import oss_get_image, oss_upload_image
 
 logger = logging.getLogger()
@@ -67,7 +67,7 @@ class LineArtService:
 
     @staticmethod
     def line_art_preprocess(image):
-        img = mmcv.imread(image)
+        img = image
         ori_shape = img.shape[:2]
         img_scale_w, img_scale_h = ori_shape
         if ori_shape[0] > 1024:
@@ -76,9 +76,9 @@ class LineArtService:
             img_scale_h = 1024
         # 如果图片size任意一边 大于 1024， 则会resize 成1024
         if ori_shape != (img_scale_w, img_scale_h):
-            # mmcv.imresize(img, img_scale_h, img_scale_w) # 老代码 引以为戒！哈哈哈~ h和w写反了
+            # my_imnormalize(img, img_scale_h, img_scale_w) # 老代码 引以为戒！哈哈哈~ h和w写反了
             img = cv2.resize(img, (img_scale_h, img_scale_w))
-        img = mmcv.imnormalize(img, mean=np.array([123.675, 116.28, 103.53]), std=np.array([58.395, 57.12, 57.375]), to_rgb=True)
+        img = my_imnormalize(img, mean=np.array([123.675, 116.28, 103.53]), std=np.array([58.395, 57.12, 57.375]), to_rgb=True)
         preprocessed_img = np.expand_dims(img.transpose(2, 0, 1), axis=0)
         return preprocessed_img, ori_shape