gameui/ocr
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

将不是三通道的图像转换为三通道

Browse Source
This commit is contained in:
satori
2023-12-03 17:40:29 +08:00
parent 1e8be5dd82
commit cbcbc64899
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pp.py
View File

@@ -36,14 +36,15 @@ def download_image(url: str) -> Image.Image:
return None return None
try: try:
if url.startswith('http://image.gameuiux.cn/') or url.startswith('https://image.gameuiux.cn/'): if url.startswith('http://image.gameuiux.cn/') or url.startswith('https://image.gameuiux.cn/'):
url = url.replace('http://image.gameuiux.cn/', url = url.replace('http://image.gameuiux.cn/', '').replace('https://image.gameuiux.cn/', '')
'').replace('https://image.gameuiux.cn/', '') oss_auth = oss2.Auth(config['OSS_ACCESS_KEY_ID'], config['OSS_ACCESS_KEY_SECRET'])
oss_auth = oss2.Auth( img = Image.open(io.BytesIO(oss2.Bucket(oss_auth, f'http://{config["OSS_HOST"]}', config['OSS_BUCKET_NAME']).get_object(url).read()))
config['OSS_ACCESS_KEY_ID'], config['OSS_ACCESS_KEY_SECRET'])
return Image.open(io.BytesIO(oss2.Bucket(oss_auth, f'http://{config["OSS_HOST"]}', config['OSS_BUCKET_NAME']).get_object(url).read()))
else: else:
response = requests.get(url) response = requests.get(url)
return Image.open(io.BytesIO(response.content)) img = Image.open(io.BytesIO(response.content))
if img.mode != 'RGB'
img = img.convert('RGB')
return img
except Exception as e: except Exception as e:
print(f'图片从{url}下载失败，错误信息为：{e}') print(f'图片从{url}下载失败，错误信息为：{e}')
return None return None