cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-12 14:32:38 +02:00
Code Issues Packages Projects Releases Wiki Activity

Clean some stuff that should have been in the polarization channel, get ready for pytorch's move to xy-style meshgrid

Browse Source
This commit is contained in:
Abe Levitan
2022-06-08 11:11:46 -07:00
parent 54ddb078e1
commit c3fef28314
8 changed files with 13 additions and 486 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CDTools/tools/image_processing/image_processing.py
+2 -2
View File
@@ -39,7 +39,7 @@ def centroid(im, dims=2):
"""
# For some reason this needs to be a list
indices = [t.arange(im.shape[-dims+i]).to(t.float32) for i in range(dims)]
indices = t.meshgrid(*indices)
indices = t.meshgrid(*indices, indexing='ij')
use_dims = [-dims+i for i in range(dims)]
divisor = t.sum(im, dim=use_dims)
@@ -102,7 +102,7 @@ def sinc_subpixel_shift(im, shift):
i = t.arange(im.shape[0]) - im.shape[0]//2
j = t.arange(im.shape[1]) - im.shape[1]//2
I,J = t.meshgrid(i,j)
I,J = t.meshgrid(i,j, indexing='ij')
I = 2 * np.pi * I.to(t.float32) / im.shape[0]
J = 2 * np.pi * J.to(t.float32) / im.shape[1]
I = I.to(dtype=im.dtype,device=im.device)