python为图片数据集做数据扩充
时间: 2024-03-13 16:48:16 浏览: 105
data_pipeline_python_数据集扩充_
5星 · 资源好评率100%
在Python中,可以使用PIL库(Python Imaging Library)来进行图片数据集的数据扩充。PIL库提供了一些常用的图像处理方法,可以通过这些方法来对图片进行扩充,包括旋转、翻转、裁剪、缩放等。
以下是一个简单的示例,展示如何使用PIL库对图片进行旋转和翻转操作:
```python
from PIL import Image
import numpy as np
# 读取图片
img = Image.open('example.jpg')
# 将图片转为数组
img_array = np.array(img)
# 对图片进行旋转
img_rotate = img.rotate(45)
# 对图片进行水平翻转
img_flip = img.transpose(Image.FLIP_LEFT_RIGHT)
# 将处理后的图片保存
img_rotate.save('example_rotate.jpg')
img_flip.save('example_flip.jpg')
```
除了上述的旋转和翻转操作,还可以使用PIL库进行更多的图像处理操作,例如裁剪、缩放、仿射变换等。需要根据具体的需求选择相应的方法来进行数据扩充。
阅读全文