Повернуть изображение в сравнении скорости CV2 Python

Question

Я использовал 2 разных способа поворота изображения из этого вопроса . Не зная этого, я потратил большую часть недели, пытаясь выяснить, почему мой генератор keras так сильно замедлился, поэтому я подумал, что поделюсь.

Первый поворот изображения, который я пробовал, был довольно медленным.

import time
t0 = time.time()
img1= ndimage.rotate(img1, 210)
t1 = time.time()

total1 = t1-t0
print(total1)

Затем я попробовал этот поворот:

def rotate(img, angle):
    row, col, channel = img.shape
    rotation_point = (row / 2, col / 2)
    rotation_matrix = cv2.getRotationMatrix2D(rotation_point, angle, 1)
    rotated_img = cv2.warpAffine(img, rotation_matrix, (col, row))
    return rotated_img 

t0 = time.time()
img1 = rotate(img1, 210)
t1 = time.time()

total2 = t1-t0
print('def: ', str(total2))
print(total2/total1)

Answer 1

Когда я использовал второй способ поворота изображения, он дал мне 14-кратное ускорение, что сэкономило мне много времени и, надеюсь, поможет и вам.