Skip to content

Commit

Permalink
FIX allow any character in mask name (#533)
Browse files Browse the repository at this point in the history
  • Loading branch information
mvdoc committed Apr 20, 2024
1 parent 35ebd43 commit 2e654e0
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions cortex/dataset/braindata.py
@@ -1,10 +1,12 @@
import hashlib
from copy import deepcopy
import numpy as np

import h5py
import numpy as np

from ..database import db


class BrainData(object):
"""
Abstract base class for brain data.
Expand Down Expand Up @@ -592,17 +594,18 @@ def blend_curvature(self, alpha, threshold=0, brightness=0.5,


def _find_mask(nvox, subject, xfmname):
import glob
import os
import re
import glob

import nibabel
files = db.get_paths(subject)['masks'].format(xfmname=xfmname, type="*")
for fname in glob.glob(files):
nib = nibabel.load(fname)
mask = nib.get_fdata().T != 0
if nvox == np.sum(mask):
fname = os.path.split(fname)[1]
name = re.compile(r'mask_([\w]+).nii.gz').search(fname)
name = re.compile(r'mask_(.+).nii.gz').search(fname)
return name.group(1), mask

raise ValueError('Cannot find a valid mask')
Expand Down

0 comments on commit 2e654e0

Please sign in to comment.