def load(self, directory, atlas_name): if not os.path.isdir(directory): print "Error: Atlas directory", directory, "does not exist or is not a directory." raise "<cluster.py> I/O error" fname_base = os.path.join(directory,atlas_name) fname_atlas = fname_base+'.p' fname_polydata = fname_base+'.vtp' if not os.path.exists(fname_atlas): print "Error: Atlas file", fname_atlas, "does not exist." raise "<cluster.py> I/O error" if not os.path.exists(fname_polydata): print "Error: Atlas file", fname_polydata, "does not exist." raise "<cluster.py> I/O error" atlas = pickle.load(open(fname_atlas,'rb')) atlas.nystrom_polydata = io.read_polydata(fname_polydata) print "<cluster.py> Atlas loaded. Nystrom polydata sample:", atlas.nystrom_polydata.GetNumberOfLines(), \ "Atlas size:", atlas.pinv_A.shape, "number of eigenvectors:", atlas.number_of_eigenvectors atlas.polydata_filename = fname_polydata # check for any version issues if not hasattr(atlas, 'version'): atlas.version = '0.0' # this is an atlas from pre 1.0 version code. Set to default values. if not hasattr(atlas, 'distance_method'): atlas.distance_method = 'Hausdorff' if not hasattr(atlas, 'bilateral'): atlas.bilateral = 'False' print "Loaded atlas", atlas_name, "from", directory #print atlas pprint (vars(atlas)) return(atlas)
def load(self, directory, atlas_name): fname = os.path.join(directory,atlas_name) atlas = pickle.load(open(fname+'.p','rb')) atlas.nystrom_polydata = io.read_polydata(fname+'.vtp') return(atlas)