import cv2
import numpy as np
import os
import glob
# 数据增强函数
def augment_data(img):
rows,cols,_ = img.shape
# 水平翻转图像
if np.random.random() > 0.5:
img = cv2.flip(img, 1)
img_name = os.path.splitext(save_path)[0] + "_flip.png"
cv2.imwrite(img_name, img)
print("Saved augmented image:", img_name)
# 随机缩放图像
scale = np.random.uniform(0.9, 1.1)
M = cv2.getRotationMatrix2D((cols/2, rows/2), 0, scale)
img_transformed = cv2.warpAffine(img, M, (cols, rows))
img_name = os.path.splitext(save_path)[0] + "_transform.png"
cv2.imwrite(img_name, img_transformed)
print("Saved augmented image:", img_name)
# 随机旋转图像
angle = np.random.randint(-10, 10)
M = cv2.getRotationMatrix2D((cols/2, rows/2), angle, 1)
img_rotated = cv2.warpAffine(img, M, (cols, rows))
img_name = os.path.splitext(save_path)[0] + "_rotated.png"
cv2.imwrite(img_name, img_rotated)
print("Saved augmented image:", img_name)
# 添加高斯噪音
mean = 0
std = np.random.uniform(5, 15)
noise = np.zeros(img.shape, np.float32)
cv2.randn(noise, mean, std)
noise = np.uint8(noise)
img_noisy = cv2.add(img, noise)
img_name = os.path.splitext(save_path)[0] + "_noisy.png"
cv2.imwrite(img_name, img_noisy)
print("Saved augmented image:", img_name)
# 随机调整对比度和亮度
alpha = np.random.uniform(0.8, 1.2)
beta = np.random.randint(-10, 10)
img_contrast = cv2.convertScaleAbs(img, alpha=alpha, beta=beta)
img_name = os.path.splitext(save_path)[0] + "_contrast.png"
cv2.imwrite(img_name, img_contrast)
print("Saved augmented image:", img_name)
return img
# 读取 data 文件夹中的所有图片,并进行数据增强
data_dir = "data"
save_dir = "result"
if not os.path.exists(save_dir):
os.makedirs(save_dir)
# 使用 glob 库来遍历 data 文件夹中所有图像
for img_path in glob.glob(os.path.join(data_dir, "*.png")):
img = cv2.imread(img_path)
# 获取保存增强后的图片文件名
img_name = os.path.basename(img_path)
save_path = os.path.join(save_dir, img_name)
# 数据增强
augment_data(img)
# 保存原始图片
cv2.imwrite(save_path, img)
print("Saved original image:", save_path)
YOLO图像前处理及格式转换
Richard Yue2024-03-30 21:28
相关推荐
繁依Fanyi9 分钟前
使用 Spring Boot + Redis + Vue 实现动态路由加载页面星尘安全16 分钟前
一种新的电子邮件攻击方式:AiTM尘浮生24 分钟前
Java项目实战II基于Java+Spring Boot+MySQL的洗衣店订单管理系统(开发文档+源码+数据库)鸽芷咕26 分钟前
【Python报错已解决】xlrd.biffh.XLRDError: Excel xlsx file; not supported铁匠匠匠1 小时前
【C总集篇】第八章 数组和指针Unicorn建模1 小时前
2024“华为杯”中国研究生数学建模竞赛(E题)深度剖析|数学建模完整过程+详细思路+代码全解析编程小白煎堆1 小时前
C语言:枚举类型秋邱1 小时前
C++: 类和对象(上)好多吃的啊1 小时前
背景图鼠标放上去切换图片过渡效果神奇夜光杯1 小时前
Python酷库之旅-第三方库Pandas(123)