Skip to content

Commit ea23e61

Browse files
committed
add par for seed fa
1 parent 751a040 commit ea23e61

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

src/cardiotensor/tractography/generate_streamlines.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,7 @@ def generate_streamlines_from_params(
254254
end_xyz: tuple[int | None, int | None, int | None] = (None, None, None),
255255
bin_factor: int = 1,
256256
num_seeds: int = 20000,
257+
fa_seed_min: float = 0.4,
257258
fa_threshold: float = 0.1,
258259
step_length: float = 0.5,
259260
max_steps: int | None = None,
@@ -341,7 +342,9 @@ def generate_streamlines_from_params(
341342

342343
# --- Load vector field ---
343344
print("📥 Loading vector field...")
344-
vector_field = DataReader(vec_load_dir).load_volume(
345+
346+
vec_reader = DataReader(vec_load_dir)
347+
vector_field = vec_reader.load_volume(
345348
start_index=start_z_binned, end_index=end_z_binned
346349
)[:, :, start_y_binned:end_y_binned, start_x_binned:end_x_binned]
347350

@@ -376,7 +379,7 @@ def generate_streamlines_from_params(
376379
)
377380
fa_volume = fa_volume[:, start_y_binned:end_y_binned, start_x_binned:end_x_binned]
378381

379-
seed_mask = fa_volume > (fa_threshold * 255)
382+
seed_mask = fa_volume > (fa_seed_min * 255)
380383
valid_indices = np.argwhere(seed_mask)
381384
if len(valid_indices) < num_seeds:
382385
print(

0 commit comments

Comments
 (0)