design design batch

2024-12-11 11:06:20 +08:00
parent 2c0a7729b8
commit e6f0ee7f3a
17 changed files with 281 additions and 26 deletions
--- a/app/service/design_batch/utils/design_ensemble.py
+++ b/app/service/design_batch/utils/design_ensemble.py
@@ -85,7 +85,7 @@ def seg_preprocess(img_path):
    if ori_shape != (img_scale_w, img_scale_h):
        # mmcv.imresize(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 = mmcv.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