-
Notifications
You must be signed in to change notification settings - Fork 17
/
InpaintingMinimizeIntroducedEnergy.cpp
122 lines (98 loc) · 4.32 KB
/
InpaintingMinimizeIntroducedEnergy.cpp
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
116
117
118
119
120
121
122
/*=========================================================================
*
* Copyright David Doria 2012 [email protected]
*
* 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.
*
*=========================================================================*/
// ITK
#include "itkImageFileReader.h"
// Submodules
#include <Helpers/Helpers.h>
#include <ITKVTKHelpers/ITKVTKHelpers.h>
#include <Mask/MaskOperations.h>
// Boost
#include <boost/graph/grid_graph.hpp>
#include <boost/property_map/property_map.hpp>
#include "Drivers/InpaintingIntroducedEnergy.hpp"
// Run with: Data/trashcan.png Data/trashcan.mask 15 filled.png
int main(int argc, char *argv[])
{
// Verify arguments
if(argc != 6)
{
std::cerr << "Required arguments: image.png imageMask.mask patchHalfWidth numberOfKNN output.png" << std::endl;
std::cerr << "Input arguments: ";
for(int i = 1; i < argc; ++i)
{
std::cerr << argv[i] << " ";
}
return EXIT_FAILURE;
}
std::stringstream ssArguments;
for(int i = 1; i < argc; ++i)
{
ssArguments << argv[i] << " ";
std::cout << argv[i] << " ";
}
// Parse arguments
std::string imageFileName;
std::string maskFileName;
unsigned int patchHalfWidth = 0;
unsigned int numberOfKNN = 0;
unsigned int binsPerChannel = 0;
std::string outputFileName;
ssArguments >> imageFileName >> maskFileName >> patchHalfWidth >> numberOfKNN >> binsPerChannel >> outputFileName;
// Output arguments
std::cout << "Reading image: " << imageFileName << std::endl;
std::cout << "Reading mask: " << maskFileName << std::endl;
std::cout << "Patch half width: " << patchHalfWidth << std::endl;
std::cout << "numberOfKNN: " << numberOfKNN << std::endl;
std::cout << "Output: " << outputFileName << std::endl;
// typedef itk::Image<itk::CovariantVector<unsigned char, 3>, 2> OriginalImageType; // This doesn't allow for direct "a - b" pixel comparisons, because (100 - 150) or similar will underflow!
// typedef itk::Image<itk::CovariantVector<float, 3>, 2> OriginalImageType; // This is quite slow
typedef itk::Image<itk::CovariantVector<int, 3>, 2> OriginalImageType;
// typedef itk::Image<itk::CovariantVector<short, 3>, 2> OriginalImageType; // Can't use this (signed range: -32768 to 32767 (not high enough, needs to handle 255*255 = 65025) )
typedef itk::ImageFileReader<OriginalImageType> ImageReaderType;
ImageReaderType::Pointer imageReader = ImageReaderType::New();
imageReader->SetFileName(imageFileName);
imageReader->Update();
OriginalImageType* originalImage = imageReader->GetOutput();
ITKHelpers::WriteRGBImage(originalImage, "OriginalImage.png");
ITKHelpers::WriteImage(originalImage, "OriginalImage.mha");
itk::ImageRegion<2> fullRegion = originalImage->GetLargestPossibleRegion();
// ImageType::Pointer image = ImageType::New();
// ITKHelpers::DeepCopy(imageReader->GetOutput(), image.GetPointer());
// Convert the image to HSV
typedef itk::Image<itk::CovariantVector<float, 3>, 2> HSVImageType;
HSVImageType::Pointer hsvImage = HSVImageType::New();
// ITKVTKHelpers::ConvertRGBtoHSV(image.GetPointer(), hsvImage.GetPointer());
ITKVTKHelpers::ConvertRGBtoHSV(originalImage, hsvImage.GetPointer());
ITKHelpers::WriteImage(hsvImage.GetPointer(), "HSVImage.mha");
// Read the mask
Mask::Pointer mask = Mask::New();
mask->Read(maskFileName);
ITKHelpers::WriteImage(mask.GetPointer(), "Mask.mha");
InpaintingIntroducedEnergy(originalImage, mask, patchHalfWidth, numberOfKNN);
// If the output filename is a png file, then use the RGBImage writer so that it is first
// casted to unsigned char. Otherwise, write the file directly.
if(Helpers::GetFileExtension(outputFileName) == "png")
{
ITKHelpers::WriteRGBImage(originalImage, outputFileName);
}
else
{
ITKHelpers::WriteImage(originalImage, outputFileName);
}
return EXIT_SUCCESS;
}