人工智能相关公司众多,这些公司通常专注于AI技术的研发、应用和推广。以下是一些基础概念和相关信息:
假设我们遇到数据不足的问题,可以使用数据增强技术来扩充数据集。以下是一个使用imgaug
库进行图像数据增强的示例代码:
import imgaug.augmenters as iaa
from PIL import Image
import os
# 定义数据增强序列
seq = iaa.Sequential([
iaa.Fliplr(0.5), # 水平翻转
iaa.Affine(rotate=(-20, 20)), # 旋转
iaa.GaussianBlur(sigma=(0, 1.0)) # 高斯模糊
])
# 加载图像文件夹
image_folder = 'path_to_image_folder'
output_folder = 'path_to_output_folder'
# 确保输出文件夹存在
os.makedirs(output_folder, exist_ok=True)
# 遍历图像文件夹并进行数据增强
for filename in os.listdir(image_folder):
if filename.endswith('.jpg') or filename.endswith('.png'):
image = Image.open(os.path.join(image_folder, filename))
image = np.array(image)
augmented_images = seq(images=[image])
for i, aug_image in enumerate(augmented_images):
output_path = os.path.join(output_folder, f"{filename.split('.')[0]}_aug{i}.jpg")
Image.fromarray(aug_image).save(output_path)
通过这种方式,可以有效地扩充数据集,提高模型的泛化能力。
领取专属 10元无门槛券
手把手带您无忧上云