From cb4b2b4eef867d896162d5126296551083076be6 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 11 Jul 2024 17:53:27 +0800 Subject: [PATCH] =?UTF-8?q?feat=20fix=20=20=E4=BF=AE=E5=A4=8D=E5=88=86?= =?UTF-8?q?=E5=89=B2=E5=87=BA=E7=8E=B0=E7=9A=84=E7=BC=9D=E9=9A=99=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/design/utils/design_ensemble.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/app/service/design/utils/design_ensemble.py b/app/service/design/utils/design_ensemble.py index 00d391f..58f0ec9 100644 --- a/app/service/design/utils/design_ensemble.py +++ b/app/service/design/utils/design_ensemble.py @@ -78,10 +78,11 @@ def seg_preprocess(img_path): img_scale_w = 1024 if ori_shape[1] > 1024: img_scale_h = 1024 - scale_factor = [] - img, x, y = mmcv.imresize(img, (img_scale_w, img_scale_h), return_scale=True) - scale_factor.append(x) - scale_factor.append(y) + # 如果图片size任意一边 大于 1024， 则会resize 成1024 + if ori_shape != (img_scale_w, img_scale_h): + # TODO 取消代码中所有 关于mmcv的resize + # 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) preprocessed_img = np.expand_dims(img.transpose(2, 0, 1), axis=0) return preprocessed_img, ori_shape