ITK进行形态学闭运算与连通域大小筛选

最新推荐文章于 2024-05-22 19:28:52 发布

zzh__98

最新推荐文章于 2024-05-22 19:28:52 发布

阅读量120

点赞数

分类专栏： ITK 文章标签：算法 c++ 开发语言

本文链接：https://blog.csdn.net/qq_44820732/article/details/131289481

版权

ITK 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

ITK图像处理

闭运算

	using ImageType = itk::Image<short, 3>;

	// closing filter
	gLogInfo << "start closing filter" << std::endl;
	typedef itk::BinaryBallStructuringElement<ImageType::PixelType, 3> StructuringElementType;
	StructuringElementType structuringElement;
	structuringElement.SetRadius(5); // radius

	typedef itk::BinaryMorphologicalClosingImageFilter<ImageType, ImageType, StructuringElementType> ClosingFilterType;
	ClosingFilterType::Pointer closingFilter = ClosingFilterType::New();
	closingFilter->SetInput(outputImg);
	closingFilter->SetKernel(structuringElement);
	closingFilter->Update();
	gLogInfo << "after closing filter" << std::endl;

连通域筛选

	// Component filter
	gLogInfo << "start component filter" << std::endl;
	typedef itk::ConnectedComponentImageFilter<ImageType, ImageType> ConnectedComponentFilterType;
	ConnectedComponentFilterType::Pointer connectedComponentFilter = ConnectedComponentFilterType::New();
	//connectedComponentFilter->SetInput(closingFilter->GetOutput());
	connectedComponentFilter->SetInput(outputImg);
	connectedComponentFilter->Update();
	
	const unsigned int volumeThreshold = 20;

	using LabelImageType = ConnectedComponentFilterType::OutputImageType;
	// get labelMap
	using LabelImageToLabelMapFilterType = itk::LabelImageToLabelMapFilter<LabelImageType>;
	LabelImageToLabelMapFilterType::Pointer labelImageToLabelMapFilter = LabelImageToLabelMapFilterType::New();
	labelImageToLabelMapFilter->SetInput(connectedComponentFilter->GetOutput());
	labelImageToLabelMapFilter->Update();

	using LabelMapType = LabelImageToLabelMapFilterType::OutputImageType;
	LabelMapType::Pointer labelMap = labelImageToLabelMapFilter->GetOutput();
	int cnt = labelMap->GetNumberOfLabelObjects();
	labelMap->SetBackgroundValue(0);
	using LabelObjectType = itk::ShapeLabelObject<short, 3>;
	std::vector<LabelObjectType::Pointer> filtered_label_objects;
	for (unsigned int i = 0; i < labelMap->GetNumberOfLabelObjects(); ++i)
	{
		
		LabelObjectType *labelObject = static_cast<LabelObjectType *>(labelMap->GetNthLabelObject(i));
		const double area = labelObject->GetNumberOfLines();
		if (area < volumeThreshold)
		{
			//std::cout << area << std::endl;
			filtered_label_objects.push_back(labelObject);
		}
	}

	for (int i = 0; i < filtered_label_objects.size(); i++)
	{
		labelMap->RemoveLabelObject(filtered_label_objects[i]);
	}
	
	using LabelMapToLabelImageFilterType = itk::LabelMapToLabelImageFilter<LabelMapType, ImageType>;
	LabelMapToLabelImageFilterType::Pointer labelMapToLabelImageFilter = LabelMapToLabelImageFilterType::New();
	labelMapToLabelImageFilter->SetInput(labelMap);
	labelMapToLabelImageFilter->Update();
	ImageType *outputImage = labelMapToLabelImageFilter->GetOutput();
	// 输出标签不是二值？ 通过迭代器修改标签值
	itk::ImageRegionIterator<ImageType> outputIterator(outputImage, outputImage->GetLargestPossibleRegion());
	
	while (!outputIterator.IsAtEnd())
	{
		float value = (float)(outputIterator.Get());
		if (value != 0)
		{
			outputIterator.Set(1);
		}
		++outputIterator;
	}