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

使用本地图像

Browse Source
This commit is contained in:
satori
2024-11-10 17:37:14 +08:00
parent bd600ee153
commit a85f252010
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
start.sh
View File

@@ -1,7 +1,6 @@
#!/bin/sh #!/bin/sh
# 创建虚拟环境 Python 3.10.12 # 创建虚拟环境 Python 3.10.12
cd ~/reverse_image_search_gpu
python3.10 -m venv venv python3.10 -m venv venv
source venv/bin/activate source venv/bin/activate

7
utilities/download.py
View File

@@ -13,6 +13,13 @@ def download_image(url:str) -> Image:
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/'):
try: try:
url = url.replace('http://image.gameuiux.cn/', '').replace('https://image.gameuiux.cn/', '') url = url.replace('http://image.gameuiux.cn/', '').replace('https://image.gameuiux.cn/', '')
# 先尝试本地图像
if os.path.exists(f"oss/{url}"):
print('使用本地图像:', url)
return Image.open(f"{IMAGES_PATH}/{url}")
# 使用 OSS 下载
print('使用 OSS 下载:', url)
obj = bucket_image2.get_object(url).read() obj = bucket_image2.get_object(url).read()
return Image.open(io.BytesIO(obj)) return Image.open(io.BytesIO(obj))
except Exception: except Exception: