BUG: ComputeTrainingMask preserve image info. TubeExtractor count.

ComputeTrainingMask was returning an image with default image info (e.g., spacing = 1,1,1). Now copies the info from the input image. TubeExtractor was numbering tubes starting at 0 - but that made tracking tubes by their Id in a mask image problematic. Now the first extracted tubes has the Id=1.
InsightSoftwareConsortium · Dec 6, 2023 · 7325399 · 7325399
1 parent 1dc871d
commit 7325399
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/examples/Demo-ManipulateTubes-JupyterNotebook.pdf b/examples/Demo-ManipulateTubes-JupyterNotebook.pdf
diff --git a/src/Segmentation/itktubeComputeTrainingMaskFilter.hxx b/src/Segmentation/itktubeComputeTrainingMaskFilter.hxx
@@ -109,6 +109,7 @@ ComputeTrainingMaskFilter< TInputImage, TLabelMap >
 {
   typename ImageType::Pointer input = ImageType::New();
   input->Graft( const_cast< ImageType * >( this->GetInput() ) );
+  input->CopyInformation( this->GetInput() );
 
   m_Threshold->SetInput( input );
   m_Threshold->Update();

diff --git a/src/Segmentation/itktubeTubeExtractor.hxx b/src/Segmentation/itktubeTubeExtractor.hxx
@@ -586,7 +586,7 @@ TubeExtractor<TInputImage>
 
       typedef itk::MinimumMaximumImageCalculator<TubeMaskImageType> MinMaxCalcType;
       typename MinMaxCalcType::Pointer maxCalc = MinMaxCalcType::New();
-      unsigned int count = 0;
+      unsigned int count = 1;
       double successRatio = 1;
       double maxValue = m_SeedExtractionMinimumProbability;
       while( ( m_SeedMaskMaximumNumberOfPoints == 0