opencv python 输入键盘旋转任意角度图像

最新推荐文章于 2024-09-04 15:11:26 发布

原创最新推荐文章于 2024-09-04 15:11:26 发布 · 1.9k 阅读

3 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一个使用Python和OpenCV库实现的图像旋转和保存的脚本。脚本通过键盘输入控制图像的旋转角度，直到找到满意的旋转效果后进行保存。此过程在一个无限循环中进行，直至用户决定停止。

while(读取一个字符)
        读取图像
        if 字符 = r
            旋转一个角度
        else if 字符 = s
            保存图像
        else if 字符 = q
            break;

from pynput import keyboard

import cv2
import numpy as np
import re

import cv2 as cv
import numpy as np
import os
#读取所有图像 ，重命名为VOC格式，选取部分图像进行平移变换

import os.path
import glob



def rotateImg(img,angel):
    rows, cols = img.shape
    M = cv2.getRotationMatrix2D(((cols - 1) / 2.0, (rows - 1) / 2.0), angel, 1)  # 旋转中心x,旋转中心y，旋转角度，缩放因子
    img = cv2.warpAffine(img, M, (cols, rows))  #在内存里完成了旋转
    cv2.namedWindow("旋转图像")
    cv2.imshow('旋转图像', img)
    cv2.waitKey(1000)
    cv2.destroyWindow("旋转图像")




if __name__ == '__main__':

    SourceImgPath = "C:\\darknet-master\\VOCdevkit\\VOC2007\\JPEGImages" + '\\'  # 文件路径
    total_sou = os.listdir(SourceImgPath)  # 获得所有图片名字
    #旋转图像
    for img_source in total_sou:
        ImgPath = os.path.join('%s%s' % (SourceImgPath, img_source))  # 获得文件夹下所有文件的路径   读取路径和保存路径
        img1 = cv2.imread(ImgPath, 0)  #读取图像

        # cols-1 and rows-1 are the coordinate limits.
        rotate_angel = 5

        while 1:
            line = input()
            print(line[0:1])
            if (line[0:1] == 'r'):  # 寻找到合适的旋转角度
                cv2.destroyAllWindows()  # 关闭之前的窗口
                rotateImg(img1, rotate_angel)#传参是非引用
                rotate_angel += 30
            elif (line[0:1] == 's'):
                #先旋转
                rows, cols = img1.shape
                M = cv2.getRotationMatrix2D(((cols - 1) / 2.0, (rows - 1) / 2.0), rotate_angel, 1)  # 旋转中心x,旋转中心y，旋转角度，缩放因子
                img1 = cv2.warpAffine(img1, M, (cols, rows))  # 在内存里完成了旋转
                #再保存
                cv.imwrite(ImgPath, img1)
            elif (line[0:1] == 'q'):
                break  # 下一张图像