feat 测试批量上传图片

fix
This commit is contained in:
zhouchengrong
2024-07-19 15:10:28 +08:00
parent b92b1f7d6e
commit ed908d0472
13 changed files with 160 additions and 48 deletions

View File

@@ -5,6 +5,7 @@ import numpy as np
from PIL import Image
from cv2 import cvtColor, COLOR_BGR2RGBA
from app.service.utils.decorator import ClassCallRunTime
from app.service.utils.generate_uuid import generate_uuid
from ..builder import PIPELINES
from ...utils.conversion_image import rgb_to_rgba
@@ -17,6 +18,7 @@ class Split(object):
Split image into front and back layer according to the segmentation result
"""
@ClassCallRunTime
# KNet
def __call__(self, result):
try:
@@ -66,7 +68,7 @@ class Split(object):
# 创建中间图层
result_pattern_image_rgba = rgb_to_rgba(result['pattern_image'], result['mask'])
result_pattern_image_pil = Image.fromarray(cvtColor(result_pattern_image_rgba, COLOR_BGR2RGBA))
_, result['pattern_image_url'], _ = upload_png_mask(result_pattern_image_pil, f'{generate_uuid()}')
result['pattern'], result['pattern_image_url'], _ = upload_png_mask(result_pattern_image_pil, f'{generate_uuid()}')
return result
except Exception as e:
logging.warning(f"split runtime exception : {e} image_id : {result['image_id']}")