75 lines
3.3 KiB
Python
75 lines
3.3 KiB
Python
import io
|
|
import logging
|
|
from io import BytesIO
|
|
|
|
import boto3
|
|
import cv2
|
|
import numpy as np
|
|
from PIL import Image
|
|
from minio import Minio
|
|
|
|
from app.core.config import *
|
|
|
|
logger = logging.getLogger()
|
|
|
|
|
|
# 获取图片
|
|
def oss_get_image(bucket, object_name, data_type):
|
|
# cv2 默认全通道读取
|
|
image_object = None
|
|
try:
|
|
if OSS == "minio":
|
|
oss_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
|
|
image_data = oss_client.get_object(bucket_name=bucket, object_name=object_name)
|
|
else:
|
|
oss_client = boto3.client('s3', aws_access_key_id=S3_ACCESS_KEY, aws_secret_access_key=S3_AWS_SECRET_ACCESS_KEY, region_name=S3_REGION_NAME)
|
|
image_data = oss_client.get_object(Bucket=bucket, Key=object_name)['Body']
|
|
if data_type == "cv2":
|
|
image_bytes = image_data.read()
|
|
image_array = np.frombuffer(image_bytes, np.uint8) # 转成8位无符号整型
|
|
image_object = cv2.imdecode(image_array, cv2.IMREAD_UNCHANGED)
|
|
else:
|
|
data_bytes = BytesIO(image_data.read())
|
|
image_object = Image.open(data_bytes)
|
|
except Exception as e:
|
|
logger.warning(f"{OSS} | 获取图片出现异常 ######: {e}")
|
|
return image_object
|
|
|
|
|
|
def oss_upload_image(bucket, object_name, image_bytes):
|
|
req = None
|
|
try:
|
|
if OSS == "minio":
|
|
oss_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
|
|
req = oss_client.put_object(bucket_name=bucket, object_name=object_name, data=io.BytesIO(image_bytes), length=len(image_bytes), content_type='image/png')
|
|
else:
|
|
oss_client = boto3.client('s3', aws_access_key_id=S3_ACCESS_KEY, aws_secret_access_key=S3_AWS_SECRET_ACCESS_KEY, region_name=S3_REGION_NAME)
|
|
req = oss_client.put_object(Bucket=bucket, Key=object_name, Body=io.BytesIO(image_bytes), ContentType='image/png')
|
|
except Exception as e:
|
|
logger.warning(f"{OSS} | 上传图片出现异常 ######: {e}")
|
|
return req
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# url = "aida-results/result_0002186a-e631-11ee-86a6-b48351119060.png"
|
|
# url = "aida-collection-element/11523/Moodboard/f60af0d2-94c2-48f9-90ff-74b8e8a481b5.jpg"
|
|
# url = "aida-sys-image/images/female/outwear/0628000054.jpg"
|
|
# url = "aida-users/89/product_image/string-89.png"
|
|
# url = "test/845046c7-4f62-4f54-a4a9-c26d49c6969335b5b3a9-d335-4871-a46c-3cc3caf07da259629dfd1f1f555a2e2a9def7e719366.png"
|
|
# url = 'aida-users/89/relight_image/123-89.png'
|
|
# url = 'aida-users/89/relight_image/123-89.png'
|
|
# url = 'aida-users/89/relight_image/123-89.png'
|
|
# url = "aida-users/89/sketchboard/female/Dress/e6724ab7-8d3f-4677-abe0-c3e42ab7af85.jpeg"
|
|
# url = "aida-users/87/print/956614a2-7e75-4fbe-9ed0-c1831e37a2c9-4-87.png"
|
|
# url = "aida-users/89/single_logo/123-89.png"
|
|
# url = "aida-users/89/product_image/string-89.png"
|
|
url = 'aida-users/89/relight_image/123-89.png'
|
|
read_type = "PIL"
|
|
if read_type == "cv2":
|
|
img = oss_get_image(bucket=url.split('/')[0], object_name=url[url.find('/') + 1:], data_type=read_type)
|
|
cv2.imshow("", img)
|
|
cv2.waitKey(0)
|
|
else:
|
|
img = oss_get_image(bucket=url.split('/')[0], object_name=url[url.find('/') + 1:], data_type=read_type)
|
|
img.show()
|