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

feat

Browse Source 
fix  重写所有resize代码，mmcv替换为cv
This commit is contained in:
zhouchengrong
2024-07-18 10:28:26 +08:00
parent 37a303f1ec
commit ac64cdcc54
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/service/design/items/pipelines/split.py
View File

@@ -45,10 +45,15 @@ class Split(object):
rgba_image = rgb_to_rgba((result['final_image'].shape[0], result['final_image'].shape[1]), result['final_image'], result['mask'])
result_front_image = np.zeros_like(rgba_image)
result_front_image[front_mask != 0] = rgba_image[front_mask != 0]
# TODO PIL resize替换为CV2
# result_front_image_pil = Image.fromarray(cvtColor(result_front_image, COLOR_BGR2RGBA))
# front_new_size = (int(result_front_image_pil.width * result["scale"] * result["resize_scale"][0]), int(result_front_image_pil.height * result["scale"] * result["resize_scale"][1]))
# result_front_image_pil = result_front_image_pil.resize(front_new_size, Image.LANCZOS)
front_new_size = (int(result_front_image.shape[1] * result["scale"] * result["resize_scale"][0]), int(result_front_image.shape[0] * result["scale"] * result["resize_scale"][1]))
result_front_image = cv2.resize(result_front_image, front_new_size)
result_front_image_pil = Image.fromarray(cvtColor(result_front_image, COLOR_BGR2RGBA))
front_new_size = (int(result_front_image_pil.width * result["scale"] * result["resize_scale"][0]), int(result_front_image_pil.height * result["scale"] * result["resize_scale"][1]))
result_front_image_pil = result_front_image_pil.resize(front_new_size, Image.LANCZOS)
# result['front_mask_image'] = cv2.resize(front_mask, front_new_size)
# result['front_image'] = result_front_image_pil
front_mask = cv2.resize(front_mask, front_new_size)
@@ -57,10 +62,15 @@ class Split(object):
if result["name"] in ('blouse', 'dress', 'outwear', 'tops'):
result_back_image = np.zeros_like(rgba_image)
result_back_image[back_mask != 0] = rgba_image[back_mask != 0]
# TODO PIL resize替换为CV2
# result_back_image_pil = Image.fromarray(cvtColor(result_back_image, COLOR_BGR2RGBA))
# back_new_size = (int(result_back_image_pil.width * result["scale"] * result["resize_scale"][0]), int(result_back_image_pil.height * result["scale"] * result["resize_scale"][1]))
# result_back_image_pil = result_back_image_pil.resize(back_new_size, Image.LANCZOS)
back_new_size = (int(result_back_image.shape[1] * result["scale"] * result["resize_scale"][0]), int(result_back_image.shape[0] * result["scale"] * result["resize_scale"][1]))
result_back_image = cv2.resize(result_back_image, back_new_size)
result_back_image_pil = Image.fromarray(cvtColor(result_back_image, COLOR_BGR2RGBA))
back_new_size = (int(result_back_image_pil.width * result["scale"] * result["resize_scale"][0]), int(result_back_image_pil.height * result["scale"] * result["resize_scale"][1]))
result_back_image_pil = result_back_image_pil.resize(back_new_size, Image.LANCZOS)
# result['back_mask_image'] = cv2.resize(back_mask, back_new_size)
# result['back_image'] = result_back_image_pil