使用imgaug数据库对“1”文件夹中的png图像进行随机裁剪、翻转、旋转、色彩变换操作,并将其保存在“1”文件夹中
时间: 2023-06-26 18:06:27 浏览: 159
可以使用以下代码实现:
```python
import os
import cv2
import imgaug.augmenters as iaa
# 定义图像增强器
aug = iaa.Sequential([
iaa.Crop(px=(0, 16)), # 随机裁剪
iaa.Fliplr(p=0.5), # 随机水平翻转
iaa.Rotate((-45, 45)), # 随机旋转
iaa.Multiply((0.5, 1.5)), # 随机色彩变换
])
# 遍历文件夹中的所有图像文件
for filename in os.listdir("1"):
if filename.endswith(".png"):
path = os.path.join("1", filename)
# 读取图像
img = cv2.imread(path)
# 对图像进行增强
img_aug = aug(image=img)
# 保存增强后的图像
cv2.imwrite(path, img_aug)
```
这段代码首先定义了一个图像增强器 `aug`,其中包含了随机裁剪、翻转、旋转、色彩变换等操作。然后遍历文件夹中的所有png图像文件,对每个图像进行增强并保存。注意,这里直接覆盖原有的图像文件,如果需要保留原有的图像文件,请先将它们复制到另一个文件夹中。
阅读全文