Manipulation of image conversions.
#include <iomanip>
#include <stdlib.h>
#include <visp3/core/vpConfig.h>
#include <visp3/core/vpDebug.h>
#include <visp3/core/vpImage.h>
#include <visp3/core/vpImageConvert.h>
#include <visp3/core/vpIoTools.h>
#include <visp3/core/vpTime.h>
#include <visp3/io/vpImageIo.h>
#include <visp3/io/vpParseArgv.h>
#define GETOPTARGS "cdi:o:n:h"
void usage(const char *name, const char *badparam, std::string ipath, std::string opath, std::string user, int nbiter)
{
fprintf(stdout, "\n\
Test image conversions.\n\
\n\
SYNOPSIS\n\
%s [-i <input image path>] [-o <output image path>] [-n <nb benchmark iterations>]\n\
[-h]\n \
", name);
fprintf(stdout, "\n\
OPTIONS: Default\n\
-i <input image path> %s\n\
Set image input path.\n\
From this path read \"Klimt/Klimt.pgm\"\n\
and \"Klimt/Klimt.ppm\" images.\n\
Setting the VISP_INPUT_IMAGE_PATH environment\n\
variable produces the same behaviour than using\n\
this option.\n\
\n\
-o <output image path> %s\n\
Set image output path.\n\
From this directory, creates the \"%s\"\n\
subdirectory depending on the username, where \n\
Klimt_grey.pgm and Klimt_color.ppm output images\n\
are written.\n\
\n\
-n <nb benchmark iterations> %d\n\
Set the number of benchmark iterations.\n\
\n\
-h\n\
Print the help.\n\n", ipath.c_str(), opath.c_str(), user.c_str(), nbiter);
if (badparam)
fprintf(stdout, "\nERROR: Bad parameter [%s]\n", badparam);
}
bool getOptions(int argc, const char **argv, std::string &ipath, std::string &opath, const std::string &user,
int &nbIterations)
{
const char *optarg_;
int c;
switch (c) {
case 'i':
ipath = optarg_;
break;
case 'o':
opath = optarg_;
break;
case 'n':
nbIterations = atoi(optarg_);
break;
case 'h':
usage(argv[0], NULL, ipath, opath, user, nbIterations);
return false;
break;
case 'c':
case 'd':
break;
default:
usage(argv[0], optarg_, ipath, opath, user, nbIterations);
return false;
break;
}
}
if ((c == 1) || (c == -1)) {
usage(argv[0], NULL, ipath, opath, user, nbIterations);
std::cerr << "ERROR: " << std::endl;
std::cerr << " Bad argument " << optarg_ << std::endl << std::endl;
return false;
}
return true;
}
void computeRegularRGBaToGrayscale(const unsigned char *rgba, unsigned char *grey, unsigned int size)
{
const unsigned char *pt_input = rgba;
const unsigned char *pt_end = rgba + size * 4;
unsigned char *pt_output = grey;
while (pt_input != pt_end) {
*pt_output = (unsigned char)(0.2126 * (*pt_input) + 0.7152 * (*(pt_input + 1)) + 0.0722 * (*(pt_input + 2)));
pt_input += 4;
pt_output++;
}
}
void computeRegularRGBToGrayscale(const unsigned char *rgb, unsigned char *grey, unsigned int size)
{
const unsigned char *pt_input = rgb;
const unsigned char *pt_end = rgb + size * 3;
unsigned char *pt_output = grey;
while (pt_input != pt_end) {
*pt_output = (unsigned char)(0.2126 * (*pt_input) + 0.7152 * (*(pt_input + 1)) + 0.0722 * (*(pt_input + 2)));
pt_input += 3;
pt_output++;
}
}
void computeRegularBGRToGrayscale(unsigned char *bgr, unsigned char *grey, unsigned int width, unsigned int height,
bool flip)
{
int lineStep = (flip) ? -(int)(width * 3) : (int)(width * 3);
unsigned char *src = (flip) ? bgr + (width * height * 3) + lineStep : bgr;
unsigned int j = 0;
unsigned int i = 0;
for (i = 0; i < height; i++) {
unsigned char *line = src;
for (j = 0; j < width; j++) {
*grey++ = (unsigned char)(0.2126 * *(line + 2) + 0.7152 * *(line + 1) + 0.0722 * *(line + 0));
line += 3;
}
src += lineStep;
}
}
#if (VISP_HAVE_OPENCV_VERSION >= 0x020101)
{
if (src.
type() == CV_8UC3) {
dest.
resize((
unsigned int)src.rows, (
unsigned int)src.cols);
if (src.isContinuous()) {
computeRegularBGRToGrayscale((
unsigned char *)src.data, (
unsigned char *)dest.
bitmap, (
unsigned int)src.cols,
(unsigned int)src.rows, false);
}
}
}
#endif
int main(int argc, const char **argv)
{
try {
std::string env_ipath;
std::string opt_ipath;
std::string opt_opath;
std::string ipath;
std::string opath;
std::string filename;
std::string username;
int nbIterations = 100;
if (!env_ipath.empty())
ipath = env_ipath;
#if defined(_WIN32)
opt_opath = "C:/temp";
#else
opt_opath = "/tmp";
#endif
if (getOptions(argc, argv, opt_ipath, opt_opath, username, nbIterations) == false) {
exit(-1);
}
if (!opt_ipath.empty())
ipath = opt_ipath;
if (!opt_opath.empty())
opath = opt_opath;
try {
} catch (...) {
usage(argv[0], NULL, ipath, opt_opath, username, nbIterations);
std::cerr << std::endl << "ERROR:" << std::endl;
std::cerr << " Cannot create " << opath << std::endl;
std::cerr << " Check your -o " << opt_opath << " option " << std::endl;
exit(-1);
}
}
if (opt_ipath.empty()) {
if (ipath != env_ipath) {
std::cout << std::endl << "WARNING: " << std::endl;
std::cout << " Since -i <visp image path=" << ipath << "> "
<< " is different from VISP_IMAGE_PATH=" << env_ipath << std::endl
<< " we skip the environment variable." << std::endl;
}
}
if (opt_ipath.empty() && env_ipath.empty()) {
usage(argv[0], NULL, ipath, opt_opath, username, nbIterations);
std::cerr << std::endl << "ERROR:" << std::endl;
std::cerr << " Use -i <visp image path> option or set VISP_INPUT_IMAGE_PATH " << std::endl
<< " environment variable to specify the location of the " << std::endl
<< " image path where test images are located." << std::endl
<< std::endl;
exit(-1);
}
std::cout << "** Convert a grey image (.pgm) to a color image (.ppm)" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert a color image (.ppm) to a grey image (.pgm)" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert YUV pixel value to a RGB value" << std::endl;
unsigned char y = 187, u = 10, v = 30;
unsigned char r, g, b;
std::cout << " y(" << (int)y << ") u(" << (int)u << ") v(" << (int)v << ") = r(" << (int)r << ") g(" << (int)g
<< ") b(" << (int)b << ")" << std::endl;
#ifdef VISP_HAVE_OPENCV
#if VISP_HAVE_OPENCV_VERSION < 0x020408
std::cout << "** Convert an IplImage to a vpImage<vpRGBa>" << std::endl;
IplImage *image = NULL;
std::cout << " Reading the color image with opencv: " << filename << std::endl;
if ((image = cvLoadImage(filename.c_str(), CV_LOAD_IMAGE_COLOR)) == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return (-1);
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Convert result in " << filename << std::endl;
std::cout << " Reading the greyscale image with opencv: " << filename << std::endl;
if (image != NULL)
cvReleaseImage(&image);
if ((image = cvLoadImage(filename.c_str(), CV_LOAD_IMAGE_GRAYSCALE)) == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return (-1);
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Convert result in " << filename << std::endl;
std::cout << "** Convert an IplImage to a vpImage<unsigned char>" << std::endl;
std::cout << " Reading the color image with opencv: " << filename << std::endl;
if (image != NULL)
cvReleaseImage(&image);
if ((image = cvLoadImage(filename.c_str(), CV_LOAD_IMAGE_COLOR)) == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return (-1);
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Convert result in " << filename << std::endl;
std::cout << " Reading the greyscale image with opencv: " << filename << std::endl;
if (image != NULL)
cvReleaseImage(&image);
if ((image = cvLoadImage(filename.c_str(), CV_LOAD_IMAGE_GRAYSCALE)) == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return (-1);
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Convert result in " << filename << std::endl;
std::cout << "** Convert a vpImage<vpRGBa> to an IplImage" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Write " << filename << std::endl;
if ((cvSaveImage(filename.c_str(), image)) == 0) {
std::cout << " Cannot write image: " << filename << std::endl;
if (image != NULL)
cvReleaseImage(&image);
return (-1);
}
std::cout << " Convert result in " << filename << std::endl;
std::cout << "** Convert a vpImage<unsigned char> to an IplImage" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Write " << filename << std::endl;
if ((cvSaveImage(filename.c_str(), image)) == 0) {
std::cout << " Cannot write image: " << std::endl << filename << std::endl;
if (image != NULL)
cvReleaseImage(&image);
return (-1);
}
std::cout << " Convert result in " << filename << std::endl;
if (image != NULL)
cvReleaseImage(&image);
std::cout << "== Conversion c interface : " << t1 - t0 << " ms" << std::endl;
#endif
#if VISP_HAVE_OPENCV_VERSION >= 0x020100
std::cout << "** Convert a cv::Mat to a vpImage<vpRGBa>" << std::endl;
cv::Mat imageMat;
std::cout << " Reading the color image with c++ interface of opencv: " << filename << std::endl;
imageMat = cv::imread(filename, 1);
if (imageMat.data == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return -1;
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Reading the greyscale image with opencv: " << filename << std::endl;
imageMat = cv::imread(filename, 0);
if (imageMat.data == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return (-1);
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert a cv::Mat to a vpImage<nsigned char>" << std::endl;
std::cout << " Reading the color image with opencv: " << filename << std::endl;
imageMat = cv::imread(filename, 1);
if (imageMat.data == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return -1;
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Reading the greyscale image with opencv: " << filename << std::endl;
imageMat = cv::imread(filename, 0);
if (imageMat.data == NULL) {
std::cout << " Cannot read image: " << filename << std::endl;
return (-1);
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Convert result in " << filename << std::endl;
std::cout << "** Convert a vpImage<vpRGBa> to a cv::Mat" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
if (!cv::imwrite(filename, imageMat)) {
std::cout << " Cannot write image: " << filename << std::endl;
return (-1);
}
std::cout << " Convert result in " << filename << std::endl;
std::cout << "** Convert a vpImage<unsigned char> to a cv::Mat" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
if (!cv::imwrite(filename, imageMat)) {
std::cout << " Cannot write image: " << filename << std::endl;
return (-1);
}
std::cout << " Convert result in " << filename << std::endl;
std::cout << "== Conversion c++ interface : " << t3 - t2 << " ms" << std::endl;
#endif
#endif
std::cout << "** Split a vpImage<vpRGBa> to vpImage<unsigned char>" << std::endl;
std::cout << " Load " << filename << std::endl;
for (int i = 0; i < 1000; i++) {
}
std::cout << " Time for 1000 split (ms): " << endtime - begintime << std::endl;
std::cout << " Save Klimt R channel: " << filename << std::endl;
std::cout << " Save Klimt B channel: " << filename << std::endl;
std::cout << "** Merge 4 vpImage<unsigned char> (RGBa) to vpImage<vpRGBa>" << std::endl;
for (int i = 0; i < 1000; i++) {
}
std::cout << " Time for 1000 merge (ms): " << endtime - begintime << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert a vpImage<vpRGBa> in RGB color space to a "
"vpImage<vpRGBa> in HSV color"
<< std::endl;
unsigned char *hue = new unsigned char[size];
unsigned char *saturation = new unsigned char[size];
unsigned char *value = new unsigned char[size];
std::cout << " Resulting image saved in: " << filename << std::endl;
double *hue2 = new double[size];
double *saturation2 = new double[size];
double *value2 = new double[size];
unsigned char *rgba = new unsigned char[size * 4];
if (hue2 != NULL) {
delete[] hue2;
hue2 = NULL;
}
if (saturation2 != NULL) {
delete[] saturation2;
saturation2 = NULL;
}
if (value2 != NULL) {
delete[] value2;
value2 = NULL;
}
std::cout << " Resulting image saved in: " << filename << std::endl;
for (
unsigned int i = 0; i < Ic.
getHeight(); i++) {
for (
unsigned int j = 0; j < Ic.
getWidth(); j++) {
if (Ic[i][j].R != I_HSV2RGBa[i][j].R || Ic[i][j].G != I_HSV2RGBa[i][j].G || Ic[i][j].B != I_HSV2RGBa[i][j].B) {
std::cerr << "Ic[i][j].R=" << static_cast<unsigned>(Ic[i][j].R)
<< " ; I_HSV2RGBa[i][j].R=" << static_cast<unsigned>(I_HSV2RGBa[i][j].R) << std::endl;
std::cerr << "Ic[i][j].G=" << static_cast<unsigned>(Ic[i][j].G)
<< " ; I_HSV2RGBa[i][j].G=" << static_cast<unsigned>(I_HSV2RGBa[i][j].G) << std::endl;
std::cerr << "Ic[i][j].B=" << static_cast<unsigned>(Ic[i][j].B)
<< " ; I_HSV2RGBa[i][j].B=" << static_cast<unsigned>(I_HSV2RGBa[i][j].B) << std::endl;
}
}
}
std::cout << "** Construction of a vpImage from an array with copyData==true" << std::endl;
unsigned char *rgba2 = new unsigned char[size * 4];
memset(rgba2, 127, size * 4);
delete[] rgba2;
std::cout << " Resulting image saved in: " << filename << std::endl;
I_copyData[0][0].R = 10;
}
{
std::cout << "** Benchmark and test RGBa / RGB / cv::Mat to Grayscale "
"conversion"
<< std::endl;
unsigned char value_sse = 0, value_regular = 0;
for (int iteration = 0; iteration < nbIterations; iteration++) {
value_sse += I_gray_sse[0][0];
}
for (int iteration = 0; iteration < nbIterations; iteration++) {
computeRegularRGBaToGrayscale((
unsigned char *)I_color.
bitmap, I_gray_regular.bitmap, I_color.
getSize());
value_regular += I_gray_regular[0][0];
}
double rmse_error = 0.0;
for (
unsigned int i = 0; i < I_color.
getHeight(); i++) {
for (
unsigned int j = 0; j < I_color.
getWidth(); j++) {
rmse_error += (I_gray_sse[i][j] - I_gray_regular[i][j]) * (I_gray_sse[i][j] - I_gray_regular[i][j]);
}
}
std::cout << "\n RGBa to Grayscale" << std::endl;
std::cout << " t_regular (" << nbIterations << " iterations)=" << t_regular << " ms"
<< " ; t_sse (" << nbIterations << " iterations)=" << t_sse << " ms" << std::endl;
std::cout << " Speed-up=" << (t_regular / t_sse) << "X" << std::endl;
std::cout <<
" RMSE error between SSE and regular version: " << (std::sqrt(rmse_error / I_color.
getSize()))
<< std::endl;
std::cout << " value_sse=" << static_cast<unsigned>(value_sse)
<< " ; value_regular=" << static_cast<unsigned>(value_regular) << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
unsigned char *rgb_array =
new unsigned char[I_color.
getSize() * 3];
value_sse = 0;
value_regular = 0;
unsigned char *rgb2gray_array_sse =
new unsigned char[I_color.
getSize()];
for (int iteration = 0; iteration < nbIterations; iteration++) {
value_sse += rgb2gray_array_sse[0];
}
unsigned char *rgb2gray_array_regular =
new unsigned char[I_color.
getSize()];
for (int iteration = 0; iteration < nbIterations; iteration++) {
computeRegularRGBToGrayscale(rgb_array, rgb2gray_array_regular, I_color.
getSize());
value_regular += rgb2gray_array_regular[0];
}
false);
rmse_error = 0.0;
for (
unsigned int i = 0; i < I_color.
getHeight(); i++) {
for (
unsigned int j = 0; j < I_color.
getWidth(); j++) {
rmse_error +=
(I_gray2rgba_sse[i][j] - I_gray2rgba_regular[i][j]) * (I_gray2rgba_sse[i][j] - I_gray2rgba_regular[i][j]);
}
}
std::cout << "\n RGB to Grayscale" << std::endl;
std::cout << " t_regular (" << nbIterations << " iterations)=" << t_regular << " ms"
<< " ; t_sse (" << nbIterations << " iterations)=" << t_sse << " ms" << std::endl;
std::cout << " Speed-up=" << (t_regular / t_sse) << "X" << std::endl;
std::cout <<
" RMSE error between SSE and regular version: " << (std::sqrt(rmse_error / I_color.
getSize()))
<< std::endl;
std::cout << " value_sse=" << static_cast<unsigned>(value_sse)
<< " ; value_regular=" << static_cast<unsigned>(value_regular) << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
#if (VISP_HAVE_OPENCV_VERSION >= 0x020101)
std::cout << "\n BGR cv::Mat to Grayscale" << std::endl;
cv::Mat colorMat = cv::imread(filename);
std::cout << " colorMat=" << colorMat.cols << "x" << colorMat.rows << std::endl;
value_sse = 0;
value_regular = 0;
for (int iteration = 0; iteration < nbIterations; iteration++) {
value_sse += I_mat2gray_sse[0][0];
}
for (int iteration = 0; iteration < nbIterations; iteration++) {
computeRegularBGRToGrayscale(colorMat, I_mat2gray_regular);
value_regular += I_mat2gray_sse[0][0];
}
rmse_error = 0.0;
for (
unsigned int i = 0; i < I_color.
getHeight(); i++) {
for (
unsigned int j = 0; j < I_color.
getWidth(); j++) {
rmse_error +=
(I_mat2gray_sse[i][j] - I_mat2gray_regular[i][j]) * (I_mat2gray_sse[i][j] - I_mat2gray_regular[i][j]);
}
}
std::cout << " t_regular (" << nbIterations << " iterations)=" << t_regular << " ms"
<< " ; t_sse (" << nbIterations << " iterations)=" << t_sse << " ms" << std::endl;
std::cout << " Speed-up=" << (t_regular / t_sse) << "X" << std::endl;
std::cout <<
" RMSE error between SSE and regular version: " << (std::sqrt(rmse_error / I_color.
getSize()))
<< std::endl;
std::cout << " value_sse=" << static_cast<unsigned>(value_sse)
<< " ; value_regular=" << static_cast<unsigned>(value_regular) << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n BGR Mat to Grayscale Mat" << std::endl;
cv::Mat grayscaleMat(colorMat.size(), CV_8U);
unsigned char value_mat = 0;
for (int iteration = 0; iteration < nbIterations; iteration++) {
cv::cvtColor(colorMat, grayscaleMat, cv::COLOR_BGR2GRAY);
value_mat += grayscaleMat.ptr<uchar>(0)[0];
}
std::cout << " t_opencv (" << nbIterations << " iterations)=" << t_opencv << " ms"
<< " ; t_sse (" << nbIterations << " iterations)=" << t_sse << " ms" << std::endl;
std::cout << " Speed-up=" << (t_opencv / t_sse) << "X" << std::endl;
std::cout << " value_mat=" << static_cast<unsigned>(value_mat) << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n RGB to Grayscale + Flip" << std::endl;
unsigned char *rgb2gray_flip_array_sse =
new unsigned char[I_color.
getSize()];
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n Conversion BGR to Grayscale + Flip" << std::endl;
unsigned char *bgr2gray_flip_array_sse =
new unsigned char[I_color.
getSize()];
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n RGB to Grayscale + Flip + Crop" << std::endl;
cv::Rect rect_roi(11, 17, 347, 449);
cv::Mat colorMat_crop = colorMat(rect_roi);
cv::Mat colorMat_crop_continous = colorMat(rect_roi).clone();
std::cout << " colorMat_crop: " << colorMat_crop.cols << "x" << colorMat_crop.rows << " is continuous? "
<< colorMat_crop.isContinuous() << std::endl;
std::cout << " colorMat_crop_continous: " << colorMat_crop_continous.cols << "x" << colorMat_crop_continous.rows
<< " is continuous? " << colorMat_crop_continous.isContinuous() << std::endl;
(unsigned int)(rect_roi.width - rect_roi.x));
for (unsigned int i = (unsigned int)rect_roi.y; i < (unsigned int)rect_roi.height; i++) {
for (unsigned int j = (unsigned int)rect_roi.x; j < (unsigned int)rect_roi.width; j++) {
I_color_crop[(unsigned int)((int)i - rect_roi.y)][(unsigned int)((int)j - rect_roi.x)] = I_color[i][j];
}
}
std::cout << " Resulting image saved in: " << filename << std::endl;
unsigned char *rgb_array_crop =
new unsigned char[I_color_crop.
getSize() * 3];
unsigned char *rgb2gray_flip_crop_array_sse =
new unsigned char[I_color_crop.
getSize()];
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n BGR to Grayscale + Flip + Crop" << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n BGR to Grayscale + Flip + Crop + No continuous Mat" << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
delete[] rgb_array_crop;
#endif
delete[] rgb_array;
std::cout << "Test succeed" << std::endl;
}
return 0;
std::cout <<
"Catch an exception: " << e.
getMessage() << std::endl;
return 1;
}
}