feat design 提速测试

fix
This commit is contained in:
zhouchengrong
2024-09-20 16:27:26 +08:00
parent ac6a21d8c0
commit 1385fde9ce
2 changed files with 2 additions and 3 deletions

View File

@@ -30,7 +30,6 @@ class Painting(object):
resize_pattern = cv2.resize(pattern, (dim_image_w, dim_image_h), interpolation=cv2.INTER_AREA)
closed_mo = np.expand_dims(result['mask'], axis=2).repeat(3, axis=2)
gray_mo = np.expand_dims(result['gray'], axis=2).repeat(3, axis=2)
logger.info(f"image id is :{result['image_id']}")
get_image_fir = resize_pattern * (closed_mo / 255) * (gray_mo / 255)
result['pattern_image'] = get_image_fir.astype(np.uint8)
result['final_image'] = result['pattern_image']