1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/*=========================================================================
*
* Copyright UMC Utrecht and contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#include <itkElastixRegistrationMethod.h>
#include <elxParameterObject.h>
#include <Core/elxVersionMacros.h>
#include <itkImage.h>
#include <itkSize.h>
#include <cassert>
int
main()
{
try
{
enum
{
imageSizeX = 6,
imageSizeY = 5
};
std::cout << "Image registation example: estimating the translation between two images of " << imageSizeX << "x"
<< imageSizeY << " pixels\n"
<< " - Linked to elastix library version " ELASTIX_VERSION_STRING "\n - Web site: https://elastix.dev"
"\n - Source code repository: https://github.com/SuperElastix/elastix"
"\n - Discussion forum: https://github.com/SuperElastix/elastix/discussions\n\n";
using ImageType = itk::Image<float>;
const auto printImage = [](const ImageType & image) {
for (int y{}; y < imageSizeY; ++y)
{
std::cout << '\t';
for (int x{}; x < imageSizeX; ++x)
{
std::cout << image.GetPixel({ x, y }) << ' ';
}
std::cout << '\n';
}
std::cout << '\n';
};
const auto fixedImage = ImageType::New();
fixedImage->SetRegions(itk::Size<>{ imageSizeX, imageSizeY });
fixedImage->AllocateInitialized();
fixedImage->SetPixel({ 1, 1 }, 3);
fixedImage->SetPixel({ 2, 1 }, 4);
fixedImage->SetPixel({ 1, 2 }, 5);
std::cout << "Fixed image:\n";
printImage(*fixedImage);
const auto movingImage = ImageType::New();
movingImage->SetRegions(itk::Size<>{ imageSizeX, imageSizeY });
movingImage->AllocateInitialized();
movingImage->SetPixel({ 3, 2 }, 3);
movingImage->SetPixel({ 4, 2 }, 4);
movingImage->SetPixel({ 3, 3 }, 5);
std::cout << "Moving image:\n";
printImage(*movingImage);
const auto parameterObject = elx::ParameterObject::New();
parameterObject->SetParameterMap(
elx::ParameterObject::ParameterMapType{ { "ImageSampler", { "Full" } },
{ "MaximumNumberOfIterations", { "44" } },
{ "Metric", { "AdvancedNormalizedCorrelation" } },
{ "Optimizer", { "AdaptiveStochasticGradientDescent" } },
{ "Transform", { "TranslationTransform" } } });
const auto filter = itk::ElastixRegistrationMethod<ImageType, ImageType>::New();
filter->SetFixedImage(fixedImage);
filter->SetMovingImage(movingImage);
filter->SetParameterObject(parameterObject);
filter->Update();
const auto * const transformParameterObject = filter->GetTransformParameterObject();
assert(transformParameterObject != nullptr);
const auto & transformParameterMaps = transformParameterObject->GetParameterMaps();
assert(!transformParameterMaps.empty());
const auto & transformParameterMap = transformParameterMaps.front();
const auto foundTransformParameter = transformParameterMap.find("TransformParameters");
assert(foundTransformParameter != transformParameterMap.cend());
std::cout << "Estimated translation:\n\t";
for (const auto & estimatedValue : foundTransformParameter->second)
{
std::cout << estimatedValue << ' ';
}
std::cout << std::endl;
}
catch (const std::exception & stdException)
{
// An exception may occur, for instance, when a components used here was not installed when elastix was built.
std::cerr << "\nCaught exception:\n" << stdException.what() << '\n';
}
} // end main()
|
