Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
    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)
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
    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)
Exemplo n.º 5
0
 def setUp(self):
     self.datafiles = getDataFiles(DATADIR)
     ref = read_vtkpd(self.datafiles[0])
         
     self.representer = statismo.vtkStandardMeshRepresenter.Create(ref)        
Exemplo n.º 6
0
    def setUp(self):
        self.datafiles = getDataFiles(DATADIR)
        ref = read_vtkpd(self.datafiles[0])

        self.representer = statismo.vtkPolyDataRepresenter.Create(
            ref, statismo.vtkPolyDataRepresenter.RIGID)
Exemplo n.º 7
0
 def setUp(self):
     self.datafiles = getDataFiles(DATADIR)
     ref = read_vtkpd(self.datafiles[0])
         
     self.representer = statismo.vtkStandardMeshRepresenter.Create(ref)        
Exemplo n.º 8
0
 def setUp(self):
     self.datafiles = getDataFiles(DATADIR)
     ref = read_vtkpd(self.datafiles[0])
         
     self.representer = statismo.vtkPolyDataRepresenter.Create(ref, statismo.vtkPolyDataRepresenter.RIGID)