antspymm 1.5.1__py3-none-any.whl → 1.5.2__py3-none-any.whl

antspymm/mm.py

@@ -3655,8 +3655,8 @@ def efficient_tensor_fit( gtab, fit_method, imagein, maskin, diffusion_model='DT
 
 
 def efficient_dwi_fit(gtab, diffusion_model, imagein, maskin,
-                  model_params=None, bvals_to_use=None,
-                  chunk_size=10, num_threads=1, verbose=True):
+                      model_params=None, bvals_to_use=None,
+                      chunk_size=1024, num_threads=1, verbose=True):
     """
     Efficient and optionally parallelized diffusion model reconstruction using DiPy.
 
@@ -3675,7 +3675,7 @@ def efficient_dwi_fit(gtab, diffusion_model, imagein, maskin,
     bvals_to_use : list of int, optional
         Subset of b-values to use for the fit (e.g., [0, 1000, 2000]).
     chunk_size : int, optional
-        Z-axis slice chunk size.
+        Maximum number of voxels per chunk (default 1024).
     num_threads : int, optional
         Number of parallel threads.
     verbose : bool, optional
@@ -3707,23 +3707,25 @@ def efficient_dwi_fit(gtab, diffusion_model, imagein, maskin,
     img_data = imagein.numpy()
     mask = maskin.numpy().astype(bool)
     X, Y, Z, N = img_data.shape
+    inplane_size = X * Y
+
+    # Convert chunk_size from voxel count to number of slices
+    slices_per_chunk = max(1, chunk_size // inplane_size)
 
     if verbose:
         print(f"[INFO] Image shape: {img_data.shape}")
         print(f"[INFO] Using model: {diffusion_model}")
-        print(f"[INFO] Chunk size: {chunk_size} | Threads: {num_threads}")
+        print(f"[INFO] Max voxels per chunk: {chunk_size} (→ {slices_per_chunk} slices) | Threads: {num_threads}")
 
-    # Filter shells if specified
     if bvals_to_use is not None:
         bvals_to_use = set(bvals_to_use)
         sel = np.isin(gtab.bvals, list(bvals_to_use))
         img_data = img_data[..., sel]
-        gtab = gradient_table(gtab.bvals[sel], gtab.bvecs[sel])
+        gtab = gradient_table(gtab.bvals[sel], bvecs=gtab.bvecs[sel])
         if verbose:
             print(f"[INFO] Selected b-values: {sorted(bvals_to_use)}")
             print(f"[INFO] Selected volumes: {sel.sum()} / {N}")
 
-    # Choose model
     def get_model(name, gtab, **params):
         if name == 'DTI':
             return dti.TensorModel(gtab, **params)
@@ -3736,14 +3738,13 @@ def efficient_dwi_fit(gtab, diffusion_model, imagein, maskin,
 
     model = get_model(diffusion_model, gtab, **model_params)
 
-    # Output volumes initialized to zero
     FA_vol = np.zeros((X, Y, Z), dtype=np.float32)
     MD_vol = np.zeros((X, Y, Z), dtype=np.float32)
     RGB_vol = np.zeros((X, Y, Z, 3), dtype=np.float32)
     has_tensor_metrics = diffusion_model in ['DTI', 'FreeWater']
 
     def process_chunk(z_start):
-        z_end = min(Z, z_start + chunk_size)
+        z_end = min(Z, z_start + slices_per_chunk)
         local_data = img_data[:, :, z_start:z_end, :]
         local_mask = mask[:, :, z_start:z_end]
         masked_data = local_data * local_mask[..., None]
@@ -3758,8 +3759,7 @@ def efficient_dwi_fit(gtab, diffusion_model, imagein, maskin,
             return z_start, z_end, FA, MD, RGB
         return z_start, z_end, None, None, None
 
-    # Run processing
-    chunks = range(0, Z, chunk_size)
+    chunks = range(0, Z, slices_per_chunk)
     if num_threads > 1:
         with ThreadPoolExecutor(max_workers=num_threads) as executor:
             futures = {executor.submit(process_chunk, z): z for z in chunks}
@@ -4455,7 +4455,7 @@ def dwi_deterministic_tracking(
     gtab = gradient_table(bvals, bvecs=bvecs, atol=2.0 )
     if mask is None:
         mask = ants.threshold_image( fa, fa_thresh, 2.0 ).iMath("GetLargestComponent")
-    dwi_data = dwi.numpy() # dwi_img.get_fdata()
+    dwi_data = dwi.numpy()
     dwi_mask = mask.numpy() == 1
     dti_model = dti.TensorModel(gtab,fit_method=fit_method)
     if verbose:

{antspymm-1.5.1.dist-info → antspymm-1.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: antspymm
-Version: 1.5.1
+Version: 1.5.2
 Summary: multi-channel/time-series medical image processing with antspyx
 Author-email: "Avants, Gosselin, Tustison, Reardon" <stnava@gmail.com>
 License: Apache-2.0

antspymm-1.5.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+antspymm/__init__.py,sha256=DnkidUfEu3Dl0tuWNTA-9VOUkBtH_cROKiPGNNXNagU,4637
+antspymm/mm.py,sha256=NbT1IBiuEMtMoanr_8yO3kLNpSSfV0j1_155gykGOM0,526972
+antspymm-1.5.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+antspymm-1.5.2.dist-info/METADATA,sha256=3Ttc-cPytZsiNct2MRz4PZwe5JmdZzaEsEgu7hxKxvA,25939
+antspymm-1.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+antspymm-1.5.2.dist-info/top_level.txt,sha256=iyD1sRhCKzfwKRJLq5ZUeV9xsv1cGQl8Ejp6QwXM1Zg,9
+antspymm-1.5.2.dist-info/RECORD,,

antspymm-1.5.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-antspymm/__init__.py,sha256=DnkidUfEu3Dl0tuWNTA-9VOUkBtH_cROKiPGNNXNagU,4637
-antspymm/mm.py,sha256=oPHhV70IhFXCKI26rP5HgKpkb2OjqCWkqoy4cSxJXqA,526866
-antspymm-1.5.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-antspymm-1.5.1.dist-info/METADATA,sha256=_I4UmZGWLM6KkmTJc21wKW1DrjT2TY723C1jr7LA3Fw,25939
-antspymm-1.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-antspymm-1.5.1.dist-info/top_level.txt,sha256=iyD1sRhCKzfwKRJLq5ZUeV9xsv1cGQl8Ejp6QwXM1Zg,9
-antspymm-1.5.1.dist-info/RECORD,,