81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
import io
|
|
import logging
|
|
from io import BytesIO
|
|
|
|
import cv2
|
|
import numpy as np
|
|
import urllib3
|
|
from PIL import Image
|
|
from minio import Minio
|
|
|
|
from utils.minio_config import MINIO_URL, MINIO_ACCESS, MINIO_SECRET, MINIO_SECURE
|
|
|
|
minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
|
|
|
|
|
|
# 自定义 Retry 类
|
|
class CustomRetry(urllib3.Retry):
|
|
def increment(self, method=None, url=None, response=None, error=None, **kwargs):
|
|
# 调用父类的 increment 方法
|
|
new_retry = super(CustomRetry, self).increment(method, url, response, error, **kwargs)
|
|
# 打印重试信息
|
|
logger.info(f"重试连接: {method} {url}，错误: {error}，重试次数: {self.total - new_retry.total}")
|
|
return new_retry
|
|
|
|
|
|
logger = logging.getLogger()
|
|
timeout = urllib3.Timeout(connect=1, read=10.0) # 连接超时 5 秒，读取超时 10 秒
|
|
http_client = urllib3.PoolManager(
|
|
num_pools=10, # 设置连接池大小
|
|
maxsize=10,
|
|
timeout=timeout,
|
|
cert_reqs='CERT_REQUIRED', # 需要证书验证
|
|
retries=CustomRetry(
|
|
total=5,
|
|
backoff_factor=0.2,
|
|
status_forcelist=[500, 502, 503, 504],
|
|
),
|
|
)
|
|
|
|
|
|
# 获取图片
|
|
def oss_get_image(oss_client, path, data_type):
|
|
# cv2 默认全通道读取
|
|
bucket = path.split("/", 1)[0]
|
|
object_name = path.split("/", 1)[1]
|
|
image_object = None
|
|
try:
|
|
image_data = oss_client.get_object(bucket_name=bucket, object_name=object_name)
|
|
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)
|
|
if image_object.dtype == np.uint16:
|
|
image_object = (image_object / 256).astype('uint8')
|
|
else:
|
|
data_bytes = BytesIO(image_data.read())
|
|
image_object = Image.open(data_bytes)
|
|
except Exception as e:
|
|
logger.warning(f"获取图片出现异常 ######: {e}")
|
|
return image_object
|
|
|
|
|
|
def oss_upload_image(oss_client, bucket, object_name, image_bytes):
|
|
req = None
|
|
try:
|
|
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')
|
|
except Exception as e:
|
|
logger.warning(f"上传图片出现异常 ######: {e}")
|
|
return req
|
|
|
|
|
|
if __name__ == '__main__':
|
|
url = "lanecarford/refaced_image/refaced1760687023.736802.png"
|
|
read_type = "1"
|
|
img = oss_get_image(oss_client=minio_client, path=url, data_type=read_type)
|
|
if read_type == "cv2":
|
|
cv2.imshow("", img)
|
|
cv2.waitKey(0)
|
|
else:
|
|
img.show()
|
|
img.save("result.png") |