讲文件夹里的图片进行高斯滤波,使图片变得模糊,用来当做数据集非常好。
import os
import cv2
def gaussian_filter(image):
# 使用高斯滤波器对图像进行滤波处理
filtered_image = cv2.GaussianBlur(image, (5, 5), 0)
return filtered_image
def process_images(input_folder, output_folder):
# 遍历输入文件夹中的所有文件
for filename in os.listdir(input_folder):
if filename.endswith(".jpg") or filename.endswith(".png"):
# 读取图像
image_path = os.path.join(input_folder, filename)
image = cv2.imread(image_path)
# 对图像进行高斯滤波处理
filtered_image = gaussian_filter(image)
# 保存滤波后的图像到输出文件夹
output_path = os.path.join(output_folder, filename)
cv2.imwrite(output_path, filtered_image)
# 设置输入文件夹和输出文件夹的路径
input_folder = "/home/zhang/yolov5_obb_master/old_new/images"
output_folder = "/home/zhang/yolov5_obb_master/old_new/dataset"
# 调用函数进行图像处理
process_images(input_folder, output_folder)