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-19 15:30:21 +08:00
parent 33e7d884cf
commit 6d68ba1add
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,7 +2,6 @@ import concurrent.futures
import io
import cv2
from PIL import Image
from app.core.config import PRIORITY_DICT
from app.service.design.core.layer import Layer
@@ -93,11 +92,11 @@ def process_object(cfg, process_id, total):
# 上传所有图片
for layer in layers:
if 'image' in layer.keys() and layer['image'] is not None:
uploaded_images.append({'image_obj': layer['image'], 'image_url': layer['image_url']})
uploaded_images.append({'image_obj': layer['image'], 'image_url': layer['image_url'], 'image_type': 'image'})
if 'pattern_image' in layer.keys() and layer['pattern_image'] is not None:
uploaded_images.append({'image_obj': layer['pattern_image'], 'image_url': layer['pattern_image_url']})
uploaded_images.append({'image_obj': layer['pattern_image'], 'image_url': layer['pattern_image_url'], 'image_type': 'pattern_image'})
if 'mask' in layer.keys() and layer['mask'] is not None and layer['mask_url'] is not None:
uploaded_images.append({'image_obj': layer['mask'], 'image_url': layer['mask_url']})
uploaded_images.append({'image_obj': layer['mask'], 'image_url': layer['mask_url'], 'image_type': 'mask'})
# 合成
items_response['synthesis_url'] = synthesis(layers, body_size)
@@ -164,7 +163,7 @@ def process_images(images):
def upload_images(image_obj):
bucket_name = image_obj['image_url'].split("/", 1)[0]
object_name = image_obj['image_url'].split("/", 1)[1]
if isinstance(image_obj['image_obj'], Image.Image):
if image_obj['image_type'] == 'image' or image_obj['image_type'] == 'pattern_image':
image_data = io.BytesIO()
image_obj['image_obj'].save(image_data, format='PNG')
image_data.seek(0)