OpenCV Python fait pivoter l'image de X degrés autour d'un point spécifique

Question

J'ai du mal à trouver des exemples de rotation d'une image autour d'un point spécifique selon un angle spécifique (souvent très petit) en Python avec OpenCV.

C'est ce que j'ai fait jusqu'à présent, mais cela produit une image résultante très étrange, mais elle est quelque peu tournée :

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
import cv2

def rotate_image(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

En supposant que vous utilisez la version cv2, ce code trouve le centre de l'image que vous voulez faire pivoter, calcule la matrice de transformation et l'applique à l'image.

Answer 2

Ou une utilisation beaucoup plus simple SciPy

from scipy import ndimage

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

voir ici pour plus d'informations sur l'utilisation.

Answer 3

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 4

J'ai eu des problèmes avec certaines des solutions ci-dessus, pour obtenir la "bounding_box" correcte ou la nouvelle taille de l'image. Voici donc ma version

def rotation(image, angleInDegrees):
    h, w = image.shape[:2]
    img_c = (w / 2, h / 2)

    rot = cv2.getRotationMatrix2D(img_c, angleInDegrees, 1)

    rad = math.radians(angleInDegrees)
    sin = math.sin(rad)
    cos = math.cos(rad)
    b_w = int((h * abs(sin)) + (w * abs(cos)))
    b_h = int((h * abs(cos)) + (w * abs(sin)))

    rot[0, 2] += ((b_w / 2) - img_c[0])
    rot[1, 2] += ((b_h / 2) - img_c[1])

    outImg = cv2.warpAffine(image, rot, (b_w, b_h), flags=cv2.INTER_LINEAR)
    return outImg

Answer 5

La fonction cv2.warpAffine prend le paramètre shape dans l'ordre inverse : (col,row) ce que les réponses ci-dessus ne mentionnent pas. Voici ce qui a fonctionné pour moi :

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