python数据增强——颜色变换（img+json）

灰灰学姐

已于 2024-01-03 15:22:34 修改

阅读量716

点赞数 10

分类专栏：数据增强文章标签： python json 开发语言

于 2024-01-03 15:17:07 首次发布

本文链接：https://blog.youkuaiyun.com/m0_51965393/article/details/135364695

版权

本文详细描述了使用Python中的cv2库对JPEGImages目录下的图像进行颜色调整，并同步处理Annotations中的json标注文件，生成新的颜色版本和相应的标注文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import cv2
import os
import json


# 定义一个函数对图像进行颜色调整，两个参数用于颜色调整强度
def getColorImg(alpha, beta, img_path, img_write_path):
    img = cv2.imread(img_path)
    colored_img = cv2.convertScaleAbs(alpha + img * beta)
    cv2.imwrite(img_write_path, colored_img)


# 定义一个函数用于处理标注数据json文件
def getColorAnno(anno_path, anno_write_path):
    with open(anno_path, 'r') as json_file:
        annotation_data = json.load(json_file)  # 加载json数据
    # 对json进行修改数据
    with open(anno_write_path, 'w') as json_file:
        json.dump(annotation_data, json_file)  # 保存修改后的json文件


# 定义一个函数用于处理一组图像喝标注数据的颜色调整
def color(alpha, beta, img_dir, anno_dir, img_write_dir, anno_write_dir):
    if not os.path.exists(img_write_dir):
        os.makedirs(img_write_dir)  # 如果输出图像目录不存在就创建

    if not os.path.exists(anno_write_dir):
        os.makedirs(anno_write_dir)  # 如果输出标注目录不存在就创建

    img_names = os.listdir(img_dir)  # 获取输入图像目录中的图像文件列表
    for i