我本地有一个face.txt文档,里面是50条url图片链接。帮我写一段python做一下数据增强,额外再生成200张
2条回答 默认 最新
关注这种问题,问下GPT,不是分分钟钟给你秒了!
讯飞星火提供:首先,我们需要使用Python的PIL库来处理图片。如果你还没有安装这个库,可以使用以下命令安装:
pip install pillow接下来,我们将编写一个Python脚本来实现数据增强。这个脚本将读取face.txt文件中的URL链接,下载图片,对其进行旋转、翻转等操作,并将增强后的图片保存到一个新的文件夹中。最后,我们将生成200张增强后的图片。
以下是实现这个功能的Python代码:
import os import requests from PIL import Image, ImageOps # 读取face.txt文件中的URL链接 with open('face.txt', 'r') as f: urls = f.readlines() # 创建一个新的文件夹来保存增强后的图片 if not os.path.exists('augmented_images'): os.makedirs('augmented_images') # 遍历URL链接,下载图片并进行数据增强 for i, url in enumerate(urls): url = url.strip() response = requests.get(url) img = Image.open(BytesIO(response.content)) # 旋转图片 rotated_img = img.rotate(45) rotated_img.save(f'augmented_images/rotated_{i}.jpg') # 翻转图片 flipped_img = ImageOps.mirror(img) flipped_img.save(f'augmented_images/flipped_{i}.jpg') # 生成200张增强后的图片 for j in range(1, 201): angle = (j - 1) * 45 rotated_img = img.rotate(angle) rotated_img.save(f'augmented_images/rotated_{i}_{j}.jpg') flipped_img = ImageOps.mirror(img) flipped_img.save(f'augmented_images/flipped_{i}_{j}.jpg')将上述代码保存为一个Python文件(例如:data_augmentation.py),然后在命令行中运行这个脚本:
python data_augmentation.py运行完成后,你会在当前目录下看到一个名为"augmented_images"的文件夹,其中包含了原始图片和增强后的图片。
解决 无用评论 打赏 举报