使用itk的区域增长算法分割图像

最新推荐文章于 2024-08-16 10:42:35 发布

吴天德少侠

最新推荐文章于 2024-08-16 10:42:35 发布

阅读量306

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/sdhdsf132452/article/details/128240690

版权

医学图像处理1 专栏收录该内容

35 篇文章 23 订阅

订阅专栏

头文件声明

/// <summary>
	/// 基于区域增长的股骨分割
	/// </summary>
	/// <param name="ct_image">原始itk图像</param>
	/// <param name="seeds">种子点</param>
	/// <param name="lower_thresh">阈值下界</param>
	/// <param name="upper_thresh">阈值上界</param>
	/// <param name="segment_result">分割结果itk图像</param>
	void segmentByRegionGrow(
		const ImageStyle3D& ct_image,
		const std::vector<std::vector<int>> seeds,
		const int lower_thresh,
		const int upper_thresh,
		ImageStyle3D& segment_result
	);

实现

void Segmentation::segmentByRegionGrow(
	const ImageStyle3D& ct_image, 
	const std::vector<std::vector<int>> seeds, 
	const int lower_thresh, const int upper_thresh, 
	ImageStyle3D& segment_result)
{
	using ConnectedFilterType =
		itk::ConnectedThresholdImageFilter<ImageType3D, ImageType3D>;
	// 构建区域增长filter
	ConnectedFilterType::Pointer connectedThreshold = ConnectedFilterType::New();
	connectedThreshold->SetLower(lower_thresh);
	connectedThreshold->SetUpper(upper_thresh);
	connectedThreshold->SetReplaceValue(1);

	// 依次添加种子点
	for (auto item : seeds) {
		ImageType3D::IndexType index;
		index[0] = item[0];
		// std::cout << "index[0] = " << index[0] << std::endl;
		index[1] = item[1];
		// std::cout << "index[1] = " << index[1] << std::endl;
		index[2] = item[2];
		// std::cout << "index[2] = " << index[2] << std::endl;
		connectedThreshold->AddSeed(index);
	}

	connectedThreshold->SetInput(ct_image);
	try {
		connectedThreshold->Update(); // 更新fileter，必须做
	}
	catch (const itk::ExceptionObject& ex) {
		std::cout << ex.what() << std::endl;
		std::cout << "Segmentation::segmentByRegionGrow(),区域增长分割时发生异常!" << std::endl;
		throw "异常: 区域增长分割时发生异常!";
	}
	segment_result = connectedThreshold->GetOutput();
}

用法

// 股骨种子点
vector<vector<int>> guSeedList{ {398,253,221} };
// 股骨阈值上下界, 可修改的自定义参数(基于像素调整后的设置)
int guLower = 82, guUpper = 94;

// 股骨分割
ImageStyle3D segmentBone;
seger.segmentByRegionGrow(
	ctImageNii, guSeedList, guLower, guUpper, segmentBone);
// 保存结果
comm_func.saveNiftiImage(segmentBone, "./segResult.nii.gz");