m1-ENSL/CGDI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

skeleton c++

Browse Source
This commit is contained in:
David Coeurjolly 2024-01-24 10:31:45 +01:00
parent 124b0c5273
commit d70f77b434
2 changed files with 8 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
image-processing/2-BilateralFilter/c++/bilateralFilter.cpp
View File

@ -47,12 +47,13 @@ int main(int argc, char **argv)
CLI::App app{"colorTransfer"}; CLI::App app{"colorTransfer"};
std::string sourceImage; std::string sourceImage;
app.add_option("-s,--source", sourceImage, "Source image")->required()->check(CLI::ExistingFile);; app.add_option("-s,--source", sourceImage, "Source image")->required()->check(CLI::ExistingFile);;
std::string targetImage;
app.add_option("-t,--target", targetImage, "Target image")->required()->check(CLI::ExistingFile);;
std::string outputImage= "output.png"; std::string outputImage= "output.png";
app.add_option("-o,--output", outputImage, "Output image")->required(); app.add_option("-o,--output", outputImage, "Output image")->required();
unsigned int nbSteps = 3; double sigmaSpatial = 3.0;
app.add_option("-n,--nbsteps", nbSteps, "Number of sliced steps (3)"); app.add_option("--sigmaSpatial", sigmaSpatial, "Variance of the Gaussian spatial filter.");
double sigmaValues = 15.0;
app.add_option("--sigmaValues", sigmaValues, "Variance of the Gaussian value filter.");
silent = false; silent = false;
app.add_flag("--silent", silent, "No verbose messages"); app.add_flag("--silent", silent, "No verbose messages");
CLI11_PARSE(app, argc, argv); CLI11_PARSE(app, argc, argv);
@ -61,19 +62,7 @@ int main(int argc, char **argv)
int width,height, nbChannels; int width,height, nbChannels;
unsigned char *source = stbi_load(sourceImage.c_str(), &width, &height, &nbChannels, 0); unsigned char *source = stbi_load(sourceImage.c_str(), &width, &height, &nbChannels, 0);
if (!silent) std::cout<< "Source image: "<<width<<"x"<<height<<" ("<<nbChannels<<")"<< std::endl; if (!silent) std::cout<< "Source image: "<<width<<"x"<<height<<" ("<<nbChannels<<")"<< std::endl;
int width_target,height_target, nbChannels_target;
unsigned char *target = stbi_load(targetImage.c_str(), &width_target, &height_target, &nbChannels_target, 0);
if (!silent) std::cout<< "Target image: "<<width_target<<"x"<<height_target<<" ("<<nbChannels_target<<")"<< std::endl;
if ((width*height) != (width_target*height_target))
{
std::cout<< "Image sizes do not match. "<<std::endl;
exit(1);
}
if (nbChannels < 3)
{
std::cout<< "Input images must be RGB images."<<std::endl;
exit(1);
}
//Main computation //Main computation
std::vector<unsigned char> output(width*height*nbChannels); std::vector<unsigned char> output(width*height*nbChannels);
@ -107,6 +96,5 @@ int main(int argc, char **argv)
} }
stbi_image_free(source); stbi_image_free(source);
stbi_image_free(target);
exit(0); exit(0);
} }

20
image-processing/2-BilateralFilter/c++/gaussianFilter.cpp
View File

@ -47,12 +47,10 @@ int main(int argc, char **argv)
CLI::App app{"colorTransfer"}; CLI::App app{"colorTransfer"};
std::string sourceImage; std::string sourceImage;
app.add_option("-s,--source", sourceImage, "Source image")->required()->check(CLI::ExistingFile);; app.add_option("-s,--source", sourceImage, "Source image")->required()->check(CLI::ExistingFile);;
std::string targetImage;
app.add_option("-t,--target", targetImage, "Target image")->required()->check(CLI::ExistingFile);;
std::string outputImage= "output.png"; std::string outputImage= "output.png";
app.add_option("-o,--output", outputImage, "Output image")->required(); app.add_option("-o,--output", outputImage, "Output image")->required();
unsigned int nbSteps = 3; double sigma = 3.0;
app.add_option("-n,--nbsteps", nbSteps, "Number of sliced steps (3)"); app.add_option("--sigma", sigma, "Variance of the Gaussian filter");
silent = false; silent = false;
app.add_flag("--silent", silent, "No verbose messages"); app.add_flag("--silent", silent, "No verbose messages");
CLI11_PARSE(app, argc, argv); CLI11_PARSE(app, argc, argv);
@ -61,19 +59,6 @@ int main(int argc, char **argv)
int width,height, nbChannels; int width,height, nbChannels;
unsigned char *source = stbi_load(sourceImage.c_str(), &width, &height, &nbChannels, 0); unsigned char *source = stbi_load(sourceImage.c_str(), &width, &height, &nbChannels, 0);
if (!silent) std::cout<< "Source image: "<<width<<"x"<<height<<" ("<<nbChannels<<")"<< std::endl; if (!silent) std::cout<< "Source image: "<<width<<"x"<<height<<" ("<<nbChannels<<")"<< std::endl;
int width_target,height_target, nbChannels_target;
unsigned char *target = stbi_load(targetImage.c_str(), &width_target, &height_target, &nbChannels_target, 0);
if (!silent) std::cout<< "Target image: "<<width_target<<"x"<<height_target<<" ("<<nbChannels_target<<")"<< std::endl;
if ((width*height) != (width_target*height_target))
{
std::cout<< "Image sizes do not match. "<<std::endl;
exit(1);
}
if (nbChannels < 3)
{
std::cout<< "Input images must be RGB images."<<std::endl;
exit(1);
}
//Main computation //Main computation
std::vector<unsigned char> output(width*height*nbChannels); std::vector<unsigned char> output(width*height*nbChannels);
@ -107,6 +92,5 @@ int main(int argc, char **argv)
} }
stbi_image_free(source); stbi_image_free(source);
stbi_image_free(target);
exit(0); exit(0);
} }