基于图像链接的批量下载

1. 获取图像路径

1.1 给定图像链接，这是一张图像

image_url = “https://univs-news-1256833609.cos.ap-beijing.myqcloud.com/123/upload/resources/image/7467914.jpg”

1.1通过网站规则得到其想要的图像链接

image_urls = [
f"https://univs-news-1256833609.cos.ap-beijing.myqcloud.com/123/upload/resources/image/{7467914 + i*2}.jpg" for i in range(50)
]

1.2 获取传入的参数，如果传入的图像路径参数是空，则判断是否使用默认的配置(非指定想要的图像链接)

if opt.image_urls:image_urls = opt.image_urls
else:while True:input_id = int(input("The following options:\n [0]: stop running. \n [1]: use default config.\nPlease enter the correct option: "))if input_id == 0:sys.exit()  # 退出程序# breakelif input_id == 1:image_urls = [f"https://univs-news-1256833609.cos.ap-beijing.myqcloud.com/123/upload/resources/image/{7467914 + i * 2}.jpg"for i in range(50)]break  # 输入正确，跳出循环else:print("input number is wrong，please input correct.\n")

2. 通过图像路径，获取图像，并下载到本地

# 遍历下载图像id = 1for i, image_url in enumerate(image_urls):# 发起HTTP请求获取图像数据response = requests.get(image_url)if response.status_code == 200:# 将图像数据转换为PIL Image对象image_data = BytesIO(response.content)image = PILImage.open(image_data)# 可以将 image 用于展示、保存或进一步处理image.save(f"./{cache_path}/{id}_{image_url.rsplit('/', 1)[1][:-4]}.png")  # 保存图像id += 1else:# 请求不到的数据进行错误返回print(f"Failed to download image: {i}_{image_url}")

3. 完整的代码

import os
import shutil
import sys
import argparse
import requests
from PIL import Image as PILImage
from io import BytesIOdef get_opt():"""参数设置"""parse = argparse.ArgumentParser()parse.add_argument("--cache_path", type=str, default="./tmp/data", help="缓存文件路径")parse.add_argument("--image_urls", type=list, default=[], help="请求的图像链接")opt = parse.parse_args()return optdef main():# 获取配置信息opt = get_opt()# 本地缓存路径cache_path = opt.cache_path# 清除现有的缓存文件夹(如果有的话)try:shutil.rmtree(cache_path)except:pass# 创建缓存文件夹os.makedirs(cache_path)# 获取图像链接if opt.image_urls:image_urls = opt.image_urlselse:while True:input_id = int(input("The following options:\n [0]: stop running. \n [1]: use default config.\nPlease enter the correct option: "))if input_id == 0:sys.exit()  # 退出程序# breakelif input_id == 1:image_urls = [f"https://univs-news-1256833609.cos.ap-beijing.myqcloud.com/123/upload/resources/image/{7467914 + i * 2}.jpg"for i in range(50)]break  # 输入正确，跳出循环else:print("input number is wrong，please input correct.\n")# 遍历下载图像id = 1for i, image_url in enumerate(image_urls):# 发起HTTP请求获取图像数据response = requests.get(image_url)if response.status_code == 200:# 将图像数据转换为PIL Image对象image_data = BytesIO(response.content)image = PILImage.open(image_data)# 可以将 image 用于展示、保存或进一步处理image.save(f"./{cache_path}/{id}_{image_url.rsplit('/', 1)[1][:-4]}.png")  # 保存图像id += 1else:# 请求不到的数据进行错误返回print(f"Failed to download image: {i}_{image_url}")if __name__ == '__main__':main()