def test_rotate_square():
N=501
x=modified_shepp_logan((N,N,3))[:,:,1]
x=numpy.array(x)
for angle in [0.0, 10.0,45.0,90.0,150.0,180.0,210.0]:
r_rot=ref.rotate_square(x,angle)
o_rot=ocv.rotate_square(x,angle)
mssim=MSSIM(r_rot,o_rot,16)
print angle,mssim.min()
if not (numpy.min(mssim)>0.90).all():
pylab.figure()
pylab.subplot(121)
pylab.imshow(mssim)
pylab.colorbar()
pylab.subplot(122)
pylab.imshow(r_rot-o_rot)
pylab.colorbar()
pylab.show()
assert False
def test_rotate_square():
N=501
x=modified_shepp_logan((N,N,3))[:,:,1]
x=numpy.array(x)
for angle in [0.0, 10.0,45.0,90.0,150.0,180.0,210.0]:
r_rot=ref.rotate_square(x,angle)
i_rot=ispmd.rotate_square(x,angle)
mssim=MSSIM(r_rot,i_rot,16)
mssim=scipy.ndimage.median_filter(mssim,3)
print angle,mssim.min()
if not (numpy.min(mssim)>0.9).all():
pylab.figure()
pylab.subplot(121)
pylab.imshow(mssim)
pylab.colorbar()
pylab.subplot(122)
pylab.imshow(r_rot-i_rot)
pylab.colorbar()
pylab.show()
assert False