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

feat

Browse Source 
fix  design预处理 补偿白边图url遗漏
This commit is contained in:
zhouchengrong
2024-06-24 15:00:04 +08:00
parent 5077e05985
commit b4490ebb95
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/service/design_pre_processing/service.py
View File

@@ -4,12 +4,8 @@ import time
import cv2 import cv2
import numpy as np import numpy as np
import torch import torch
from minio import Minio
from pymilvus import connections, Collection
from urllib3.exceptions import ResponseError
import torch.nn.functional as F
import tritonclient.grpc as grpcclient import tritonclient.grpc as grpcclient
import io from urllib3.exceptions import ResponseError
from app.core.config import * from app.core.config import *
from app.service.design.utils.design_ensemble import get_keypoint_result from app.service.design.utils.design_ensemble import get_keypoint_result
@@ -179,14 +175,14 @@ class DesignPreprocessing:
image_bytes = cv2.imencode(".jpg", ret)[1].tobytes() image_bytes = cv2.imencode(".jpg", ret)[1].tobytes()
# image['show_image_url'] = f"{image['image_url'].split('/', 1)[0]}/{self.minio_client.put_object(image['image_url'].split('/', 1)[0], image['image_url'].split('/', 1)[1].replace('.', '-show.'), io.BytesIO(image_bytes), len(image_bytes), content_type='image/jpeg').object_name}" # image['show_image_url'] = f"{image['image_url'].split('/', 1)[0]}/{self.minio_client.put_object(image['image_url'].split('/', 1)[0], image['image_url'].split('/', 1)[1].replace('.', '-show.'), io.BytesIO(image_bytes), len(image_bytes), content_type='image/jpeg').object_name}"
bucket_name = image['image_url'].split('/', 1)[0] bucket_name = image['image_url'].split('/', 1)[0]
object_name = image['image_url'].split('/', 1)[1] object_name = image['image_url'].split('/', 1)[1].replace('.', '-show.')
oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes) oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes)
image['show_image_url'] = f"{bucket_name}/{object_name}" image['show_image_url'] = f"{bucket_name}/{object_name}"
else: else:
image_bytes = cv2.imencode(".jpg", image['obj'])[1].tobytes() image_bytes = cv2.imencode(".jpg", image['obj'])[1].tobytes()
# image['show_image_url'] = f"{image['image_url'].split('/', 1)[0]}/{self.minio_client.put_object(image['image_url'].split('/', 1)[0], image['image_url'].split('/', 1)[1].replace('.', '-show.'), io.BytesIO(image_bytes), len(image_bytes), content_type='image/jpeg').object_name}" # image['show_image_url'] = f"{image['image_url'].split('/', 1)[0]}/{self.minio_client.put_object(image['image_url'].split('/', 1)[0], image['image_url'].split('/', 1)[1].replace('.', '-show.'), io.BytesIO(image_bytes), len(image_bytes), content_type='image/jpeg').object_name}"
bucket_name = image['image_url'].split('/', 1)[0] bucket_name = image['image_url'].split('/', 1)[0]
object_name = image['image_url'].split('/', 1)[1] object_name = image['image_url'].split('/', 1)[1].replace('.', '-show.')
oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes) oss_upload_image(bucket=bucket_name, object_name=object_name, image_bytes=image_bytes)
image['show_image_url'] = f"{bucket_name}/{object_name}" image['show_image_url'] = f"{bucket_name}/{object_name}"