iradon: Use util.pad for array padding.

This depends on util.pad accepting 0s in pad_width, which has a proposed
solution in gh-634.
This commit is contained in:
Jostein Bø Fløystad
2013-07-06 20:55:53 +02:00
parent 237fb989b0
commit ac460a9777
+4 -5
View File
@@ -109,11 +109,10 @@ def radon(image, theta=None, circle=False):
def _sinogram_circle_to_square(sinogram):
size = int(np.ceil(np.sqrt(2) * sinogram.shape[0]))
sinogram_padded = np.zeros((size, sinogram.shape[1]))
pad = (size - sinogram.shape[0]) // 2
sinogram_padded[pad:pad + sinogram.shape[0], :] = sinogram
return sinogram_padded
diagonal = int(np.ceil(np.sqrt(2) * sinogram.shape[0]))
pad = diagonal - sinogram.shape[0]
pad_width = ((pad // 2, pad - pad // 2), (0, 0))
return util.pad(sinogram, pad_width, mode='constant', constant_values=0)
def iradon(radon_image, theta=None, output_size=None,