手机怎么打开禁止访问的网站,最新被百度收录的网站,企业网站建设三个原则,中国建设银行app下载官网在目标检测任务中#xff0c;数据集的质量直接影响到模型的性能。数据增强作为提升数据集多样性和模型泛化能力的常用手段#xff0c;在图像处理过程中扮演着重要角色。锐化#xff08;Sharpening#xff09;技术是常见的图像增强方法之一#xff0c;能够突出图像中的细节…在目标检测任务中数据集的质量直接影响到模型的性能。数据增强作为提升数据集多样性和模型泛化能力的常用手段在图像处理过程中扮演着重要角色。锐化Sharpening技术是常见的图像增强方法之一能够突出图像中的细节特征使模型在处理边缘、纹理等细节时表现更好。然而单纯对图像进行锐化处理并不充分特别是在目标检测任务中图像的锐化还需与标签标注框的同步处理相结合以保证增强后的图像与其对应的标注信息保持一致性。
本篇文章将介绍如何对目标检测数据集中的图片进行锐化处理并同步调整相应的标签文件确保在数据增强的同时图像与标签信息的精确对应。 代码如下
import os,shutil
from PIL import Image, ImageEnhancedef write_xml(folder_path,output_folder, new_suffix):if not os.path.exists(output_folder):os.makedirs(output_folder)# 遍历文件夹中的所有文件for filename in os.listdir(folder_path):# 获取文件的扩展名file_base_name, file_extension os.path.splitext(filename)# 构造新的文件名new_filename f{file_base_name}{new_suffix}{file_extension}# 获取完整的原文件路径和新文件路径old_file_path os.path.join(folder_path, filename)new_file_path os.path.join(output_folder, new_filename)# 复制或重命名文件shutil.copy(old_file_path, new_file_path)# 调整对比度
def adjust_contrast(image, factor):enhancer ImageEnhance.Contrast(image)adjusted_image enhancer.enhance(factor)return adjusted_image# 调整锐度
def adjust_sharpness(image, factor):enhancer ImageEnhance.Sharpness(image)adjusted_image enhancer.enhance(factor)return adjusted_image# 批量处理文件夹中的图片
def process_images(input_folder, output_folder, sharpness_factor, new_suffix):if not os.path.exists(output_folder):os.makedirs(output_folder)for filename in os.listdir(input_folder):if filename.endswith((.jpg, .jpeg, .png, .bmp, .tif)):image_path os.path.join(input_folder, filename)image Image.open(image_path)# 调整锐度image_with_adjusted_sharpness adjust_sharpness(image, sharpness_factor)file_base_name, file_extension os.path.splitext(filename)new_filename f{file_base_name}{new_suffix}{file_extension}# 保存到输出文件夹output_path os.path.join(output_folder, new_filename)image_with_adjusted_sharpness.save(output_path)print(fProcessed and saved: {output_path})# 输入文件夹和输出文件夹路径
input_folder_jpg rE:\peanut_data\jj #输入图片文件夹
input_folder_xml rE:\peanut_data\xx #输入标签文件夹
output_folder_jpg rE:\peanut_data\jj- #输出图片文件夹
output_folder_xml rE:\peanut_data\xx- #输出标签文件夹
new_suffix _r
# 调整参数
sharpness_factor 4.0 #锐度因子# 执行批量处理
process_images(input_folder_jpg, output_folder_jpg, sharpness_factor,new_suffix)
write_xml(input_folder_xml,output_folder_xml, new_suffix)
