OpenCV Python xoay hình ảnh theo độ X xung quanh điểm cụ thể

Question

Tôi đang gặp khó khăn khi tìm các ví dụ để xoay hình ảnh xung quanh một điểm cụ thể theo góc cụ thể (thường rất nhỏ) trong Python bằng cách sử dụng OpenCV.

Đây là những gì tôi có cho đến nay, nhưng nó tạo ra một hình ảnh kết quả rất lạ, nhưng nó có thể xoay một chút:

def rotateImage(image, angle): 
    if image != None: 
     dst_image = cv.CloneImage(image) 

     rotate_around = (0,0) 
     transl = cv.CreateMat(2, 3, cv.CV_32FC1) 

     matrix = cv.GetRotationMatrix2D(rotate_around, angle, 1.0, transl) 
     cv.GetQuadrangleSubPix(image, dst_image, transl) 
     cv.GetRectSubPix(dst_image, image, rotate_around) 

    return dst_image 

Answer 1

import numpy as np 

def rotateImage(image, angle): 
    image_center = tuple(np.array(image.shape[1::-1])/2) 
    rot_mat = cv2.getRotationMatrix2D(image_center, angle, 1.0) 
    result = cv2.warpAffine(image, rot_mat, image.shape[1::-1], flags=cv2.INTER_LINEAR) 
    return result 

Giả sử bạn đang sử dụng phiên bản CV2, mã mà tìm trung tâm của hình ảnh bạn muốn xoay, tính toán ma trận chuyển đổi và áp dụng cho hình ảnh.

Answer 2

Chỉnh sửa nhanh câu trả lời @ alex-rodrigues ... đề cập đến hình dạng bao gồm số lượng kênh.

import cv2 
import numpy as np 

def rotateImage(image, angle): 
    center=tuple(np.array(image.shape[0:2])/2) 
    rot_mat = cv2.getRotationMatrix2D(center,angle,1.0) 
    return cv2.warpAffine(image, rot_mat, image.shape[0:2],flags=cv2.INTER_LINEAR) 

Answer 3

Hoặc sử dụng dễ dàng hơn nhiều SciPy

from scipy import ndimage 

#rotation angle in degree 
rotated = ndimage.rotate(image_to_rotate, 45) 

thấy here để biết thêm thông tin cách sử dụng.

Answer 4

Hàm cv2.warpAffine lấy tham số hình dạng theo thứ tự ngược lại: (col, hàng) mà các câu trả lời ở trên không đề cập đến. Đây là những gì làm việc cho tôi:

import numpy as np 

def rotateImage(image, angle): 
    row,col = image.shape 
    center=tuple(np.array([row,col])/2) 
    rot_mat = cv2.getRotationMatrix2D(center,angle,1.0) 
    new_image = cv2.warpAffine(image, rot_mat, (col,row)) 
    return new_image 

Answer 5

def rotate(image, angle, center = None, scale = 1.0): 
    (h, w) = image.shape[:2] 

    if center is None: 
     center = (w/2, h/2) 

    # Perform the rotation 
    M = cv2.getRotationMatrix2D(center, angle, scale) 
    rotated = cv2.warpAffine(image, M, (w, h)) 

    return rotated 

Answer 6

import imutils 

vs = VideoStream(src=0).start() 
... 

while (1): 
    frame = vs.read() 
    ... 

    frame = imutils.rotate(frame, 45) 

thêm: https://github.com/jrosebr1/imutils