python数据增强——镜像翻转（jpg)

最新推荐文章于 2024-08-09 20:02:35 发布

灰灰学姐

最新推荐文章于 2024-08-09 20:02:35 发布

阅读量616

点赞数 11

分类专栏：数据增强文章标签： python opencv 计算机视觉

本文链接：https://blog.csdn.net/m0_51965393/article/details/135369687

版权

import os
import cv2


# 水平翻转，创建一个水平镜像
def h_MirrorImg(img_path, img_write_path):
    img = cv2.imread(img_path)
    mirror_img = cv2.flip(img, 1)
    cv2.imwrite(img_write_path, mirror_img)


# 垂直翻转，创建一个垂直镜像
def v_MirrorImg(img_path, img_write_path):
    img = cv2.imread(img_path)
    mirror_img = cv2.flip(img, 0)
    cv2.imwrite(img_write_path, mirror_img)


# 同时进行水平和垂直翻转的图像
def a_MirrorImg(img_path, img_write_path):
    img = cv2.imread(img_path)
    mirror_img = cv2.flip(img, -1)
    cv2.imwrite(img_write_path, mirror_img)


def mirror(img_dir, img_write_dir):
    if not os.path.exists(img_write_dir):
        os.makedirs(img_write_dir)

    img_names = os.listdir(img_dir)
    for img_name in img_names:
        img_path = os.path.join(img_dir, img_name)
        # 翻转后的图像文件路径
        h_img_write_path = os.path.join(img_write_dir, img_name[:-4] + '_1' + '.jpg')
        v_img_wri