python cv2 封装 拍照截取旋转翻转 为一个函数

下面的方法实现了,从摄像头获取图片,并根据条件翻转或旋转

import cv2
# flip_x, flip_y 分别决定了是否进行x轴翻转 y轴翻转
# clip = (x,y,w,h);左上角坐标及宽高 决定了要裁剪的方形区域; 不进行裁剪则全部置0
# rotate: 0不旋转; 1 顺时针转90度; 2 旋转180度; -1 逆时针旋转90度
def take_picture(flip_x=False, flip_y=False, clip=(0, 0, 100, 100), rotate=0):
	clip_x, clip_y, clip_w, clip_h = clip
    cap = cv2.VideoCapture(0)
    # print("set pic width: ", cap.set(3, 3664))
    # print("set pic height: ", cap.set(4, 2748), rotate, flip_x)
    res, frame = cap.read()
    cap.release()
    if res:
        cv2.imwrite("images/origin.png", frame)
        if flip_x:
            frame = cv2.flip(frame, 1)
        if flip_y:
            frame = cv2.flip(frame, 0)
        if rotate:
            frame = cv2.rotate(frame, cv2.ROTATE_90_CLOCKWISE if rotate == 1 else cv2.ROTATE_90_COUNTERCLOCKWISE if rotate == -1 else cv2.ROTATE_180)
        if flip_x or flip_y or rotate:
            cv2.imwrite("images/res_rotate", frame)
        if clip_x or clip_y or clip_w or clip_h:
            img = frame[clip_y: clip_y + clip_h, clip_x: clip_x + clip_w]
            cv2.imwrite("images/res_clip.png", img)
        else:
            cv2.imwrite("images/res_clip.png.png", frame)
        return True
    else:
        return False

猜你喜欢

转载自blog.csdn.net/downanddusk/article/details/125421888