Skip to content

Commit 7e11d3a

Browse files
committed
fix: some more syntax and computing fixes for noise extraction
1 parent 64ea721 commit 7e11d3a

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

examples/rsfmri_conn_preprocessing.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -217,14 +217,17 @@ def extract_noise_components(realigned_file, mask_file, num_components=5,
217217
imgseries = nb.load(realigned_file)
218218
components = None
219219
for filename in filename_to_list(mask_file):
220-
mask = nb.load(filename)
221-
voxel_timecourses = imgseries.get_data()[np.nonzero(mask)]
222-
voxel_timecourses = voxel_timecourses.byteswap().newbyteorder()
220+
mask = nb.load(filename).get_data()
221+
voxel_timecourses = imgseries.get_data()[mask > 0]
223222
voxel_timecourses[np.isnan(np.sum(voxel_timecourses, axis=1)), :] = 0
224223
# remove mean and normalize by variance
225224
# voxel_timecourses.shape == [nvoxels, time]
226225
X = voxel_timecourses.T
227-
X = (X - np.mean(X, axis=0))/np.std(X, axis=0)
226+
stdX = np.std(X, axis=0)
227+
stdX[stdX == 0] = 1.
228+
stdX[np.isnan(stdX)] = 1.
229+
stdX[np.isinf(stdX)] = 1.
230+
X = (X - np.mean(X, axis=0))/stdX
228231
u, _, _ = sp.linalg.svd(X, full_matrices=False)
229232
if components is None:
230233
components = u[:, :num_components]

0 commit comments

Comments
 (0)