为了实现这个需求,我们可以使用?requests
?库来下载图片,并使用?PIL
?(Python Imaging Library) 来对图片进行一些简单的增强操作,如旋转、裁剪、翻转等。
首先,确保你已经安装了必要的库:
pip install requests pillow
然后,你可以使用以下代码来读取?face.txt
?中的URL,下载图片,并进行数据增强:
?import requests from PIL import Image import io import random import string import os # 下载图片并保存到指定目录 def download_image(url, save_dir): response = requests.get(url) img = Image.open(io.BytesIO(response.content)) save_path = os.path.join(save_dir, f"img_{os.getpid()}.jpg") img.save(save_path) return save_path # 数据增强函数 - 随机旋转角度 def data_augmentation(image_path, save_dir): img = Image.open(image_path) angle = random.randint(0, 360) # 随机旋转0-360度 img_rotated = img.rotate(angle) save_path = os.path.join(save_dir, f"rotated_{angle}.jpg") img_rotated.save(save_path) return save_path # 生成随机字符串作为文件名前缀,以避免文件名冲突 def generate_random_string(length): return ''.join(random.choices(string.ascii_letters + string.digits, k=length)) # 主程序 def main(): url_file = "face.txt" # 替换为你的文件路径 save_dir = "enhanced_images" # 保存增强后的图片的目录,确保它存在 if not os.path.exists(save_dir): os.makedirs(save_dir) with open(url_file, "r") as f: for url in f: # 下载图片并保存到指定目录 image_path = download_image(url.strip(), save_dir) # 数据增强 - 随机旋转图片并保存新的图片 random_str = generate_random_string(5) # 生成一个5字符的随机字符串作为前缀,避免文件名冲突 rotated_image_path = data_augmentation(image_path, save_dir) print(f"Original Image: {image_path}, Rotated Image: {rotated_image_path}") # 你还可以添加其他的数据增强操作,例如翻转、裁剪等。 # ... # os.remove(image_path) # 如果你想删除原图,可以取消注释这一行。 if __name__ == "__main__": main()
这段代码首先从?face.txt?文件中读取URL,下载图片,并将其保存到指定的目录。然后,它会对每张图片进行数据增强(在这里是随机旋转),并保存增强的图片。你可以根据需要添加更多的数据增强操作。
来源:6547网 http://6547.cn/blog/460