ITK使用elastix库作配准

#include <iostream>
#include "elastixlib.h"
#include "itkParameterFileParser.h"
#include "itkImageFileWriter.h"
#include "itkImageFileReader.h"
int main() {
	const unsigned int Dimension = 3;
	typedef short PixelType;
	typedef itk::Image<PixelType, Dimension> ImageType;
	typedef itk::ImageFileReader<ImageType> ReaderType;
	ReaderType::Pointer fixReader = ReaderType::New();
	ReaderType::Pointer movReader = ReaderType::New();
	fixReader->SetFileName("E:/registration/input/fix.mhd");
	movReader->SetFileName("E:/registration/input/mov.mhd");
	try
	{
		fixReader->Update();
		movReader->Update();
	}
	catch (itk::ExceptionObject &ex)
	{
		std::cout << ex << std::endl;
		return 0;
	}
	using namespace elastix;
	typedef ELASTIX::ParameterMapType RegistrationParametersType;
	typedef itk::ParameterFileParser ParserType;
	// Create parser for transform parameters text file.
	ParserType::Pointer file_parser = ParserType::New();
	// Try parsing transform parameters text file.
	file_parser->SetParameterFileName("E:/registration/input/parameters.txt");
	try
	{
		file_parser->ReadParameterFile();
	}
	catch (itk::ExceptionObject & e)
	{
		std::cout << e.what() << std::endl;
		// Do some error handling!
	}
	// Retrieve parameter settings as map.
	RegistrationParametersType parameters = file_parser->GetParameterMap();
	typedef std::vector<RegistrationParametersType> RegistrationParametersContainerType;
	ELASTIX* elastix = new ELASTIX();
	int error = 0;
	try
	{
		error = elastix->RegisterImages(
			static_cast<typename itk::DataObject::Pointer>(fixReader->GetOutput()),
			static_cast<typename itk::DataObject::Pointer>(movReader->GetOutput()),
			parameters, // Parameter map read in previous code
			"E:/registration/output", // Directory where output is written, if enabled
			true, // Enable/disable writing of elastix.log
			true, // Enable/disable output to console
			0, // Provide fixed image mask (optional, 0 = no mask)
			0 // Provide moving image mask (optional, 0 = no mask)
		);
	}
	catch (itk::ExceptionObject &err)
	{
		// Do some error handling.
	}
	if (error == 0)
	{
		if (elastix->GetResultImage().IsNotNull())
		{
			// Typedef the ITKImageType first...
			ImageType * output_image = static_cast<ImageType *>(
				elastix->GetResultImage().GetPointer());
		}
		else
		{
			// Registration failure. Do some error handling.
		}
		// Get transform parameters of all registration steps.
		RegistrationParametersContainerType transform_parameters
			= elastix->GetTransformParameterMapList();
		// Clean up memory.
		delete elastix;
		return 0;
	}
}

注:
1、如果图像像素类型与elastix配置文本中像素类型不一致时,会导致图像读取结果为空。
2、图像转换我也搞定了,但是重采样图像一次要五分钟- -?!

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值