def testLoadSaveSurrogateData(self): datamanager = statismo.DataManagerWithSurrogates_vtkPD.Create( self.representer, os.path.join(DATADIR, "..", "hand_images", "surrogates", "hand_surrogates_types.txt")) ds_filename = os.path.join(DATADIR, "hand-1.vtk") ds = read_vtkpd(ds_filename) surrogate_filename = os.path.join(DATADIR, "..", "hand_images", "surrogates", "hand-1_surrogates.txt") datamanager.AddDatasetWithSurrogates(ds, ds_filename, surrogate_filename) tmpfile = tempfile.mktemp(suffix="h5") datamanager.Save(tmpfile) datamanager_new = statismo.DataManagerWithSurrogates_vtkPD.Load( tmpfile) self.assertEqual(datamanager.GetNumberOfSamples(), datamanager_new.GetNumberOfSamples()) sampleSet = datamanager.GetSampleDataStructure() newSampleSet = datamanager_new.GetSampleDataStructure() for (sample, newSample) in zip(sampleSet, newSampleSet): self.assertTrue( (sample.GetSampleVector() == newSample.GetSampleVector() ).all() == True)
def setUp(self): self.datafiles = getDataFiles(DATADIR) ref = read_vtkpd(self.datafiles[0]) self.representer = statismo.vtkPolyDataRepresenter.Create(ref, statismo.vtkPolyDataRepresenter.RIGID) self.dataManager = statismo.DataManager_vtkPD.Create(self.representer) datasets = map(read_vtkpd, self.datafiles) for (dataset, filename) in zip(datasets, self.datafiles): self.dataManager.AddDataset(dataset, filename)
def setUp(self): self.datafiles = getDataFiles(DATADIR) ref = read_vtkpd(self.datafiles[0]) self.representer = statismo.vtkStandardMeshRepresenter.Create(ref) self.dataManager = statismo.DataManager_vtkPD.Create(self.representer) datasets = map(read_vtkpd, self.datafiles) for (dataset, filename) in zip(datasets, self.datafiles): self.dataManager.AddDataset(dataset, filename)
def testLoadSaveSurrogateData(self): datamanager = statismo.DataManagerWithSurrogates_vtkPD.Create(self.representer, os.path.join(DATADIR, "..", "hand_images", "surrogates", "hand_surrogates_types.txt")) ds_filename = os.path.join(DATADIR, "hand-1.vtk") ds = read_vtkpd(ds_filename) surrogate_filename = os.path.join(DATADIR, "..", "hand_images", "surrogates", "hand-1_surrogates.txt") datamanager.AddDatasetWithSurrogates(ds, ds_filename, surrogate_filename) tmpfile = tempfile.mktemp(suffix="h5") datamanager.Save(tmpfile) datamanager_new = statismo.DataManagerWithSurrogates_vtkPD.Load(tmpfile) self.assertEqual(datamanager.GetNumberOfSamples(), datamanager_new.GetNumberOfSamples()) sampleSet = datamanager.GetSampleDataStructure() newSampleSet = datamanager_new.GetSampleDataStructure() for (sample, newSample) in zip(sampleSet, newSampleSet): self.assertTrue((sample.GetSampleVector() == newSample.GetSampleVector()).all() == True)
def setUp(self): self.datafiles = getDataFiles(DATADIR) ref = read_vtkpd(self.datafiles[0]) self.representer = statismo.vtkStandardMeshRepresenter.Create(ref)
def setUp(self): self.datafiles = getDataFiles(DATADIR) ref = read_vtkpd(self.datafiles[0]) self.representer = statismo.vtkPolyDataRepresenter.Create( ref, statismo.vtkPolyDataRepresenter.RIGID)
def setUp(self): self.datafiles = getDataFiles(DATADIR) ref = read_vtkpd(self.datafiles[0]) self.representer = statismo.vtkPolyDataRepresenter.Create(ref, statismo.vtkPolyDataRepresenter.RIGID)