bf: properly handle long lesion seg with different spaces

freesurfer · Feb 13, 2024 · 3f76770 · 3f76770
1 parent b585fb9
commit 3f76770
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/samseg/SamsegLongitudinalLesion.py b/samseg/SamsegLongitudinalLesion.py
@@ -144,11 +144,11 @@ def constructTimepointModels(self):
                 wmSearchString=self.wmSearchString,
                 sampler=self.sampler
             ))
-            self.timepointModels[timepointNumber].mask = self.sstModel.mask
+            self.timepointModels[timepointNumber].mask = self.masks[timepointNumber]
             self.timepointModels[timepointNumber].imageBuffers = self.imageBuffersList[timepointNumber]
-            self.timepointModels[timepointNumber].voxelSpacing = self.sstModel.voxelSpacing
-            self.timepointModels[timepointNumber].transform = self.sstModel.transform
-            self.timepointModels[timepointNumber].cropping = self.sstModel.cropping
+            self.timepointModels[timepointNumber].voxelSpacing = self.voxelSpacings[timepointNumber]
+            self.timepointModels[timepointNumber].transform = self.transforms[timepointNumber]
+            self.timepointModels[timepointNumber].cropping = self.croppings[timepointNumber]
 
     def initializeLatentVariables(self):