📄 itkregiongrow2dtest.cxx
字号:
1.0, 9.0, 6.5, 6.5, 6.5, 6.5, 6.5, 6.5, 9.0, 1.0,
1.0, 9.0, 6.5, 6.5, 30.0, 30.0, 6.5, 6.5, 9.0, 1.0,
1.0, 9.0, 6.5, 6.5, 6.5, 6.5, 6.5, 6.5, 9.0, 1.0,
1.0, 9.0, 9.0, 9.0, 9.0, 9.0, 9.0, 9.0, 9.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0 };
typedef ImageType::PixelType::VectorType ImageData;
ImageData pixelData;
int k = 0;
while ( inIt != inItEnd ) {
pixelData[0] = inImageVals[k];
pixelData[1] = 100 - pixelData[0];
inIt.Set( pixelData );
++inIt;
++k;
}
// FIRST TEST:
// If lambda is negative, the number of final regions should equal
// initial number of regions, the region labels should be consecutive
std::cout << std::endl << "First test, lambda = -1" << std::endl;
KLMFilter->SetMaximumLambda( -1 );
// Kick off the Region grow function
LOCAL_TEST_EXCEPTION_MACRO( KLMFilter );
KLMFilter->Print(std::cout);
// This should return unique integer labels of the segmented regions.
// The region labels should be consecutive integers beginning with 1.
if( numPixels != KLMFilter->GetNumberOfRegions() )
{
std::cout << "Test FAILED" << std::endl;
return EXIT_FAILURE;
}
// Test the functions useful to test the region and border statistics
// as the regions are merged. Primarily useful for debug operations and are
// called several times, so prudent usage is advisable.
// KLMFilter->PrintAlgorithmRegionStats();
// KLMFilter->PrintAlgorithmBorderStats();
std::cout << "Extracting and checking approximation image" << std::endl;
OutputImageType::Pointer outImage = KLMFilter->GetOutput();
typedef itk::ImageRegionIterator< OutputImageType > OutputImageIterator;
inIt.GoToBegin();
OutputImageIterator outIt( outImage, outImage->GetBufferedRegion() );
typedef OutputImageType::PixelType::VectorType OutputImageData;
ImageData pixelIn;
OutputImageData pixelOut;
while ( inIt != inItEnd )
{
pixelOut = outIt.Get();
pixelIn = inIt.Get();
if ( pixelOut[0] != pixelIn[0] || pixelOut[1] != pixelIn[1] )
{
std::cout << "Test FAILED" << std::endl;
if ( pixelOut[0] != pixelIn[0] )
{
std::cout << "pixelOut[0]: " << pixelOut[0]
<< " != "
<< "pixelIn[0]: " << pixelIn[0]
<< std::endl;
}
if ( pixelOut[1] != 0 )
{
std::cout << "pixelOut[1]: " << pixelOut[1]
<< " != "
<< "pixelIn[1]: " << pixelIn[1]
<< std::endl;
}
return EXIT_FAILURE;
}
++outIt;
++inIt;
} //end while iterator loop
// Make sure that the labelled image type is set to unsigned integer
// as labels associated with different regions are always integers
std::cout << "Extracting and checking label image" << std::endl;
typedef itk::Image<LabelType, NUMDIM3D> LabelledImageType;
LabelledImageType::Pointer labelledImage = KLMFilter->GetLabelledImage();
// Loop through the approximation image and check if they match the
// input image
// setup the iterators
typedef LabelledImageType::PixelType LabelledImagePixelType;
typedef itk::ImageRegionIterator< LabelledImageType > LabelImageIterator;
LabelImageIterator
labelIt( labelledImage, labelledImage->GetBufferedRegion() );
LabelType pixelLabel;
LabelType m = 1;
while(!labelIt.IsAtEnd())
{
pixelLabel = labelIt.Get();
if(pixelLabel != m)
{
std::cout << "Test FAILED" << std::endl;
std::cout << "pixelLabel: " << pixelLabel
<< " != "
<< "m: " << m
<< std::endl;
return EXIT_FAILURE;
}
++labelIt;
++m;
} //end while iterator loop
std::cout << "Test PASSED" << std::endl;
KLMFilter->ResetPipeline();
// SECOND TEST:
// the key test for union borders
std::cout << std::endl <<
"Second test, key merging test containing duplicate borders" << std::endl;
KLMFilter->SetMaximumLambda( 1e45 );
unsigned int nregions = 8;
KLMFilter->SetMaximumNumberOfRegions( nregions );
// Kick off the Region grow function
LOCAL_TEST_EXCEPTION_MACRO( KLMFilter );
KLMFilter->Print(std::cout);
// This should return unique integer labels of the segmented regions.
// The region labels should be consecutive integers beginning with 1.
if( nregions != KLMFilter->GetNumberOfRegions() )
{
std::cout << "Test FAILED" << std::endl;
return EXIT_FAILURE;
}
// Test the functions useful to test the region and border statistics
// as the regions are merged. Primarily useful for debug operations and are
// called several times, so prudent usage is advisable.
// KLMFilter->PrintAlgorithmRegionStats();
// KLMFilter->PrintAlgorithmBorderStats();
std::cout << "Extracting and checking approximation image" << std::endl;
OutputImageType::Pointer outImage2 = KLMFilter->GetOutput();
OutputImageIterator outIt2( outImage2, outImage2->GetBufferedRegion() );
OutputImageIterator outItEnd2 = outIt2.End();
k = 0;
while ( outIt2 != outItEnd2 )
{
pixelOut = outIt2.Get();
if ( pixelOut[0] != outImageVals[k] ||
pixelOut[1] != 100 - pixelOut[0] )
{
std::cout << "Test FAILED" << std::endl;
if ( pixelOut[0] != outImageVals[k] )
{
std::cout << "pixelOut[0]: " << pixelOut[0]
<< " != "
<< "outImageVals[k]: " << outImageVals[k]
<< std::endl;
}
if ( pixelOut[1] != (100 - pixelOut[0]) )
{
std::cout << "pixelOut[1]: " << pixelOut[1]
<< " != "
<< "100 - pixelOut[0]: " << 100 - pixelOut[0]
<< std::endl;
}
return EXIT_FAILURE;
}
++outIt2;
++k;
} //end while iterator loop
std::cout << "Extracting and checking label image" << std::endl;
LabelledImageType::Pointer labelledImage2 = KLMFilter->GetLabelledImage();
LabelImageIterator
labelIt2( labelledImage2, labelledImage2->GetBufferedRegion() );
k = 0;
while(!labelIt2.IsAtEnd())
{
pixelLabel = labelIt2.Get();
if(pixelLabel != labelVals[k])
{
std::cout << "Test FAILED" << std::endl;
std::cout << "pixelLabel: " << pixelLabel
<< " != "
<< "labelVals[k]: " << labelVals[k]
<< std::endl;
return EXIT_FAILURE;
}
++labelIt2;
++k;
} //end while iterator loop
std::cout << "Test PASSED" << std::endl;
KLMFilter->ResetPipeline();
// THIRD TEST:
// degenerate case, all 0 image, all constant lambda values -
// issues related to speed of algorithm
std::cout << std::endl << "Third test, all 0 image" << std::endl;
inIt.GoToBegin();
pixelData[0] = 0;
pixelData[1] = 0;
while ( inIt != inItEnd ) {
inIt.Set( pixelData );
++inIt;
}
gridSize[0] = 1;
gridSize[1] = 2;
gridSize[2] = 1;
KLMFilter->SetInput( image );
KLMFilter->SetMaximumNumberOfRegions( 75 );
KLMFilter->SetGridSize( gridSize );
KLMFilter->SetMaximumLambda( 1e45 );
// Kick off the Region grow function
LOCAL_TEST_EXCEPTION_MACRO( KLMFilter );
// KLMFilter->Print(std::cout);
if( KLMFilter->GetMaximumNumberOfRegions() !=
KLMFilter->GetNumberOfRegions() )
{
std::cout << "Test FAILED" << std::endl;
return EXIT_FAILURE;
}
// Test the functions useful to test the region and border statistics
// as the regions are merged. Primarily useful for debug operations and are
// called several times, so prudent usage is advisable.
// KLMFilter->PrintAlgorithmRegionStats();
// KLMFilter->PrintAlgorithmBorderStats();
std::cout << "Extracting and checking approximation image" << std::endl;
OutputImageType::Pointer outImage3 = KLMFilter->GetOutput();
OutputImageIterator outIt3( outImage3, outImage3->GetBufferedRegion() );
OutputImageIterator outItEnd3 = outIt3.End();
while ( outIt3 != outItEnd3 )
{
pixelOut = outIt3.Get();
if ( pixelOut[0] != 0 || pixelOut[1] != 0 )
{
std::cout << "Test FAILED" << std::endl;
if ( pixelOut[0] != 0 )
{
std::cout << "pixelOut[0]: " << pixelOut[0]
<< " != "
<< "0"
<< std::endl;
}
if ( pixelOut[1] != 0 )
{
std::cout << "pixelOut[1]: " << pixelOut[1]
<< " != "
<< "0"
<< std::endl;
}
return EXIT_FAILURE;
}
++outIt3;
} //end while iterator loop
std::cout << "Extracting and checking label image" << std::endl;
typedef itk::Statistics::ScalarImageToHistogramGenerator<LabelledImageType>
HistogramGeneratorType;
HistogramGeneratorType::Pointer histogramGenerator =
HistogramGeneratorType::New();
histogramGenerator->SetInput( KLMFilter->GetLabelledImage() );
histogramGenerator->SetNumberOfBins( KLMFilter->GetNumberOfRegions() );
histogramGenerator->SetMarginalScale( 1.0 );
histogramGenerator->Compute();
typedef HistogramGeneratorType::HistogramType HistogramType;
const HistogramType * histogram = histogramGenerator->GetOutput();
const unsigned int histogramSize = histogram->Size();
if ( histogramSize != KLMFilter->GetNumberOfRegions() )
{
std::cout << "Test FAILED" << std::endl;
return EXIT_FAILURE;
}
HistogramType::ConstIterator histIt = histogram->Begin();
HistogramType::ConstIterator histItEnd = histogram->End();
double Sum = histogram->GetTotalFrequency();
double labelEntropy = 0.0;
while( histIt != histItEnd )
{
double probability = histIt.GetFrequency() / Sum;
if( probability == 0 )
{
std::cout << "Test FAILED" << std::endl;
return EXIT_FAILURE;
}
labelEntropy -= probability * log( probability );
++histIt;
}
labelEntropy /= log( 2.0 );
double idealEntropy = - log( 8.0/ numPixels ) / log( 2.0 );
std::cout << "Label entropy = " << labelEntropy << " bits " << std::endl;
std::cout << "Ideal entropy = " << idealEntropy << " bits " << std::end⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -