aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat

Browse Source 
fix  修复分割出现的缝隙问题
This commit is contained in:
zhouchengrong
2024-07-11 17:53:27 +08:00
parent e99026ff2a
commit cb4b2b4eef
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/service/design/utils/design_ensemble.py
View File

@@ -78,10 +78,11 @@ def seg_preprocess(img_path):
img_scale_w = 1024 img_scale_w = 1024
if ori_shape[1] > 1024: if ori_shape[1] > 1024:
img_scale_h = 1024 img_scale_h = 1024
scale_factor = [] # 如果图片size任意一边 大于 1024， 则会resize 成1024
img, x, y = mmcv.imresize(img, (img_scale_w, img_scale_h), return_scale=True) if ori_shape != (img_scale_w, img_scale_h):
scale_factor.append(x) # TODO 取消代码中所有 关于mmcv的resize
scale_factor.append(y) # 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) preprocessed_img = np.expand_dims(img.transpose(2, 0, 1), axis=0)
return preprocessed_img, ori_shape return preprocessed_img, ori_shape