Анаморфоз в питоне

Question

Я попытался сделать анаморфоз на изображении, перейдя по этой ссылке

https://github.com/aydal/Cylinderical-Anamorphosis/blob/master/anamorph.py

Это дает анаморфное изображение, но оно дает это изображение в полукруге. Но я хочу вывод в полном размере круга. Я пробовал с

warp[c-j, i-1] = img[p-1, q-1]
warp[c+j, i-1] = img[p-1, q-1]

Вместо warp[c-j, i-1] = img[p-1, q-1]

Но это не дает одно изображение в полном круге, а создает один и тот же вывод дважды!

Кто-нибудь может мне помочь, пожалуйста.

Полный код:

import math
from cv2 import *
import numpy as np

img = imread("test.jpg")
(rows, cols) = (img.shape[0], img.shape[1])
r = 0  #offset-gives space to keep cylinder and height of the image from bottom: original: math.trunc(.25*rows)
c = rows  #this will be the decisive factor in size of output image-maximum radius of warped image: original: c = r+rows
warp = np.zeros([c,2*c,3], dtype=np.uint8)


def convert(R, b):
    return math.trunc(b*rows/(2*math.asin(1))), math.trunc(c-R)


for i in range(0, 2*c):
    for j in range(1, c):
        b = math.atan2(j, i-c)
        R = math.sqrt(j*j+math.pow(i-c, 2))
        if R>=r and R<=c:
            (q, p) = convert(R, b)
            warp[c-j, i-1] = img[p-1, q-1]
            #warp[c+j, i-1] = img[p-1, q-1]

imshow("Output", warp)
waitKey()

Исходное изображение

Мое выходное изображение (полукруг)

Желаемое выходное изображение

Answer 1

Подобно смещению столбца, вы должны также включить смещение для строк при вычислении b а также R, Поскольку искаженное изображение имеет c строк смещение c//2:

b = math.atan2(j - c//2, i-c)
R = math.sqrt((j - c//2)**2 + math.pow(i-c, 2))

Обратите внимание, что деформированное изображение не является идеальным кругом, так как вы указали, что оно в два раза больше ширины. Если вы хотите полный круг, вы также должны настроить проверку верхней границы для R быть c//2 так как это макс. радиус вдоль рядов:

if r <= R <= c//2:
    ...

И так же вам нужно настроить вычисления в convert:

return ..., math.trunc(c//2 - R)

Но тогда, в любом случае, вы можете просто использовать квадратное изображение с самого начала, то есть указать warp.shape == (c, c),

редактировать

Обновленный код, использует оригинальные размеры для деформированного изображения:

import math
import cv2
import numpy as np

img = cv2.imread("/tmp/img.jpg")
(rows, cols) = (img.shape[0], img.shape[1])
r = 0
c = rows // 2
warp = np.zeros([rows, cols, 3], dtype=np.uint8)


def convert(R, b):
    return math.trunc(c * (1 - b/math.pi)), 2*math.trunc(c - R) - 1


for i in range(0, cols):
    for j in range(0, rows):
        b = math.atan2(j - c, i - c)
        R = math.sqrt((j - c)**2 + (i - c)**2)
        if r <= R <= c:
            q, p = convert(R, b)
            warp[j, i] = img[p, q]

cv2.imshow("Output", warp)
cv2.waitKey()

И выводим изображение: