图像翻转180度 脚本

import os
import cv2
import argparse

def reverse_img(input, output):
    for filename in os.listdir(input):
        file, suffix = os.path.splitext(filename)
        if suffix.lower() not in [".jpg", ".jpeg", ".png"]:
            continue
        img_path = os.path.join(input, filename)
        image = cv2.imread(img_path)
        (h, w) = image.shape[:2]
        center = (w // 2, h // 2)
        M = cv2.getRotationMatrix2D(center, 180, 1.0)
        rotated = cv2.warpAffine(image, M, (w, h))
        print(rotated)
        save_path = os.path.join(output, file + "r.jpg")
        cv2.imwrite(save_path, rotated)


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--input_dir", default="./train/")
    parser.add_argument("--out_dir", default="./train_reversed/")
    return parser.parse_args()


def main(args):
    if not os.path.isdir(args.out_dir):
        os.makedirs(args.out_dir)
    reverse_img(args.input_dir, args.out_dir)


if __name__ == "__main__":
    main(get_args())