Анаморфоз в питоне - PullRequest
Новая
       90

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

0 голосов
/ 20 января 2019

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

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()

Исходное изображение enter image description here

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

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

1 Ответ

0 голосов
/ 20 января 2019

Подобно смещению столбца, вы должны также включить смещение для строк при вычислении 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()

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

Output image

...