问题
I'm trying to do a polar transform on the first image below and end up with the second. However my result is the third image. I have a feeling it has to do with what location I choose as my "origin" but am unsure.
radius = sqrt(width**2 + height**2)
nheight = int(ceil(radius)/2)
nwidth = int(ceil(radius/2))
for y in range(0, height):
for x in range(0, width):
t = int(atan(y/x))
r = int(sqrt(x**2+y**2)/2)
color = getColor(getPixel(pic, x, y))
setColor( getPixel(radial,r,t), color)
回答1:
There are a few differences / errors:
- They use the centre of the image as the origin
- They scale the axis appropriately. In your example, you're plotting your angle (between 0 and in your case, pi), instead of utilising the full height of the image.
- You're using the wrong atan function (atan2 works a lot better in this situation :))
- Not amazingly important, but you're rounding unnecessarily quite a lot, which throws off accuracy a little and can slow things down.
This is the code combining my suggested improvements. It's not massively efficient, but it should hopefully work :)
maxradius = sqrt(width**2 + height**2)/2
rscale = width / maxradius
tscale = height / (2*math.pi)
for y in range(0, height):
dy = y - height/2
for x in range(0, width):
dx = x - width/2
t = atan2(dy,dx)%(2*math.pi)
r = sqrt(dx**2+dy**2)
color = getColor(getPixel(pic, x, y))
setColor( getPixel(radial,int(r*rscale),int(t*tscale)), color)
In particular, it fixes the above problems in the following ways:
- We use
dx = x - width / 2
as a measure of distance from the centre, and similarly withdy
. We then use these in replace ofx
,y
throughout the computation. - We will have our
r
satisfying0 <= r <= sqrt( (width/2)^2 +(height/2)^2 )
, and ourt
eventually satisfying0 < t <= 2 pi
so, I create the appropriate scale factors to putr
andt
along thex
andy
axes respectively. - Normal
atan
can only distinguish based on gradients, and is computationally unstable near vertical lines... Instead,atan2
(see http://en.wikipedia.org/wiki/Atan2) solves both problems, and accepts(y,x)
pairs to give an angle.atan2
returns an angle-pi < t <= pi
, so we can find the remainder modulo2 * math.pi
to it to get it in the range0 < t <= 2pi
ready for scaling. - I've only rounded at the end, when the new pixels get set.
Any questions, just ask!
来源:https://stackoverflow.com/questions/27080577/converting-cartesian-image-to-polar-appearance-differences