merged 2.4 into trunk
This commit is contained in:
@@ -46,86 +46,75 @@
|
||||
using namespace cv;
|
||||
using namespace std;
|
||||
|
||||
namespace cvtest
|
||||
{
|
||||
|
||||
string fourccToString(int fourcc)
|
||||
{
|
||||
return format("%c%c%c%c", fourcc & 255, (fourcc >> 8) & 255, (fourcc >> 16) & 255, (fourcc >> 24) & 255);
|
||||
}
|
||||
|
||||
const VideoFormat g_specific_fmt_list[] =
|
||||
{
|
||||
VideoFormat("avi", CV_FOURCC('X', 'V', 'I', 'D')),
|
||||
VideoFormat("avi", CV_FOURCC('M', 'P', 'E', 'G')),
|
||||
VideoFormat("avi", CV_FOURCC('M', 'J', 'P', 'G')),
|
||||
//VideoFormat("avi", CV_FOURCC('I', 'Y', 'U', 'V')),
|
||||
VideoFormat("mkv", CV_FOURCC('X', 'V', 'I', 'D')),
|
||||
VideoFormat("mkv", CV_FOURCC('M', 'P', 'E', 'G')),
|
||||
VideoFormat("mkv", CV_FOURCC('M', 'J', 'P', 'G')),
|
||||
|
||||
VideoFormat("mov", CV_FOURCC('m', 'p', '4', 'v')),
|
||||
VideoFormat()
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
class CV_HighGuiTest : public cvtest::BaseTest
|
||||
{
|
||||
protected:
|
||||
void ImageTest(const string& dir);
|
||||
void VideoTest (const string& dir, int fourcc);
|
||||
void VideoTest (const string& dir, const cvtest::VideoFormat& fmt);
|
||||
void SpecificImageTest (const string& dir);
|
||||
void SpecificVideoFileTest (const string& dir, const char codecchars[4]);
|
||||
void SpecificVideoCameraTest (const string& dir, const char codecchars[4]);
|
||||
void SpecificVideoTest (const string& dir, const cvtest::VideoFormat& fmt);
|
||||
|
||||
public:
|
||||
CV_HighGuiTest();
|
||||
~CV_HighGuiTest();
|
||||
CV_HighGuiTest() {}
|
||||
~CV_HighGuiTest() {}
|
||||
virtual void run(int) = 0;
|
||||
};
|
||||
|
||||
class CV_ImageTest : public CV_HighGuiTest
|
||||
{
|
||||
public:
|
||||
CV_ImageTest();
|
||||
~CV_ImageTest();
|
||||
CV_ImageTest() {}
|
||||
~CV_ImageTest() {}
|
||||
void run(int);
|
||||
};
|
||||
|
||||
class CV_SpecificImageTest : public CV_HighGuiTest
|
||||
{
|
||||
public:
|
||||
CV_SpecificImageTest();
|
||||
~CV_SpecificImageTest();
|
||||
CV_SpecificImageTest() {}
|
||||
~CV_SpecificImageTest() {}
|
||||
void run(int);
|
||||
};
|
||||
|
||||
class CV_VideoTest : public CV_HighGuiTest
|
||||
{
|
||||
public:
|
||||
CV_VideoTest();
|
||||
~CV_VideoTest();
|
||||
CV_VideoTest() {}
|
||||
~CV_VideoTest() {}
|
||||
void run(int);
|
||||
};
|
||||
|
||||
class CV_SpecificVideoFileTest : public CV_HighGuiTest
|
||||
class CV_SpecificVideoTest : public CV_HighGuiTest
|
||||
{
|
||||
public:
|
||||
CV_SpecificVideoFileTest();
|
||||
~CV_SpecificVideoFileTest();
|
||||
CV_SpecificVideoTest() {}
|
||||
~CV_SpecificVideoTest() {}
|
||||
void run(int);
|
||||
};
|
||||
|
||||
class CV_SpecificVideoCameraTest : public CV_HighGuiTest
|
||||
{
|
||||
public:
|
||||
CV_SpecificVideoCameraTest();
|
||||
~CV_SpecificVideoCameraTest();
|
||||
void run(int);
|
||||
};
|
||||
|
||||
CV_HighGuiTest::CV_HighGuiTest() {}
|
||||
CV_HighGuiTest::~CV_HighGuiTest() {}
|
||||
|
||||
CV_ImageTest::CV_ImageTest() : CV_HighGuiTest() {}
|
||||
CV_VideoTest::CV_VideoTest() : CV_HighGuiTest() {}
|
||||
CV_SpecificImageTest::CV_SpecificImageTest() : CV_HighGuiTest() {}
|
||||
CV_SpecificVideoFileTest::CV_SpecificVideoFileTest() : CV_HighGuiTest() {}
|
||||
CV_SpecificVideoCameraTest::CV_SpecificVideoCameraTest() : CV_HighGuiTest() {}
|
||||
|
||||
CV_ImageTest::~CV_ImageTest() {}
|
||||
CV_VideoTest::~CV_VideoTest() {}
|
||||
CV_SpecificImageTest::~CV_SpecificImageTest() {}
|
||||
CV_SpecificVideoFileTest::~CV_SpecificVideoFileTest() {}
|
||||
CV_SpecificVideoCameraTest::~CV_SpecificVideoCameraTest() {}
|
||||
|
||||
double PSNR(const Mat& m1, const Mat& m2)
|
||||
{
|
||||
Mat tmp;
|
||||
absdiff( m1.reshape(1), m2.reshape(1), tmp);
|
||||
multiply(tmp, tmp, tmp);
|
||||
|
||||
double MSE = 1.0/(tmp.cols * tmp.rows) * sum(tmp)[0];
|
||||
|
||||
return 20 * log10(255.0 / sqrt(MSE));
|
||||
}
|
||||
|
||||
void CV_HighGuiTest::ImageTest(const string& dir)
|
||||
{
|
||||
@@ -141,7 +130,26 @@ void CV_HighGuiTest::ImageTest(const string& dir)
|
||||
return;
|
||||
}
|
||||
|
||||
const string exts[] = {"png", "bmp", "tiff", "jpg", "jp2", "ppm", "ras" };
|
||||
const string exts[] = {
|
||||
#ifdef HAVE_PNG
|
||||
"png",
|
||||
#endif
|
||||
#ifdef HAVE_TIFF
|
||||
"tiff",
|
||||
#endif
|
||||
#ifdef HAVE_JPEG
|
||||
"jpg",
|
||||
#endif
|
||||
#ifdef HAVE_JASPER
|
||||
"jp2",
|
||||
#endif
|
||||
#ifdef HAVE_OPENEXR
|
||||
"exr",
|
||||
#endif
|
||||
"bmp",
|
||||
"ppm",
|
||||
"ras"
|
||||
};
|
||||
const size_t ext_num = sizeof(exts)/sizeof(exts[0]);
|
||||
|
||||
for(size_t i = 0; i < ext_num; ++i)
|
||||
@@ -213,12 +221,13 @@ void CV_HighGuiTest::ImageTest(const string& dir)
|
||||
ts->set_failed_test_info(ts->OK);
|
||||
}
|
||||
|
||||
void CV_HighGuiTest::VideoTest(const string& dir, int fourcc)
|
||||
|
||||
void CV_HighGuiTest::VideoTest(const string& dir, const cvtest::VideoFormat& fmt)
|
||||
{
|
||||
string src_file = dir + "../cv/shared/video_for_test.avi";
|
||||
string tmp_name = "video.avi";
|
||||
string tmp_name = format("video_%s.%s", cvtest::fourccToString(fmt.fourcc).c_str(), fmt.ext.c_str());
|
||||
|
||||
ts->printf(ts->LOG, "reading video : %s\n", src_file.c_str());
|
||||
ts->printf(ts->LOG, "reading video : %s and converting it to %s\n", src_file.c_str(), tmp_name.c_str());
|
||||
|
||||
CvCapture* cap = cvCaptureFromFile(src_file.c_str());
|
||||
|
||||
@@ -229,6 +238,7 @@ void CV_HighGuiTest::VideoTest(const string& dir, int fourcc)
|
||||
}
|
||||
|
||||
CvVideoWriter* writer = 0;
|
||||
vector<Mat> frames;
|
||||
|
||||
for(;;)
|
||||
{
|
||||
@@ -236,13 +246,16 @@ void CV_HighGuiTest::VideoTest(const string& dir, int fourcc)
|
||||
|
||||
if (!img)
|
||||
break;
|
||||
|
||||
frames.push_back(Mat(img).clone());
|
||||
|
||||
if (writer == 0)
|
||||
{
|
||||
writer = cvCreateVideoWriter(tmp_name.c_str(), fourcc, 24, cvGetSize(img));
|
||||
writer = cvCreateVideoWriter(tmp_name.c_str(), fmt.fourcc, 24, cvGetSize(img));
|
||||
if (writer == 0)
|
||||
{
|
||||
ts->printf(ts->LOG, "can't create writer (with fourcc : %d)\n", fourcc);
|
||||
ts->printf(ts->LOG, "can't create writer (with fourcc : %d)\n",
|
||||
cvtest::fourccToString(fmt.fourcc).c_str());
|
||||
cvReleaseCapture( &cap );
|
||||
ts->set_failed_test_info(ts->FAIL_MISMATCH);
|
||||
return;
|
||||
@@ -255,8 +268,6 @@ void CV_HighGuiTest::VideoTest(const string& dir, int fourcc)
|
||||
cvReleaseVideoWriter( &writer );
|
||||
cvReleaseCapture( &cap );
|
||||
|
||||
cap = cvCaptureFromFile(src_file.c_str());
|
||||
|
||||
CvCapture *saved = cvCaptureFromFile(tmp_name.c_str());
|
||||
if (!saved)
|
||||
{
|
||||
@@ -266,25 +277,27 @@ void CV_HighGuiTest::VideoTest(const string& dir, int fourcc)
|
||||
|
||||
const double thresDbell = 20;
|
||||
|
||||
for(;;)
|
||||
for(int i = 0;; i++)
|
||||
{
|
||||
IplImage* ipl = cvQueryFrame( cap );
|
||||
IplImage* ipl1 = cvQueryFrame( saved );
|
||||
|
||||
if (!ipl || !ipl1)
|
||||
if (!ipl1)
|
||||
break;
|
||||
|
||||
Mat img(ipl);
|
||||
Mat img = frames[i];
|
||||
Mat img1(ipl1);
|
||||
|
||||
if (PSNR(img1, img) < thresDbell)
|
||||
double psnr = PSNR(img1, img);
|
||||
if (psnr < thresDbell)
|
||||
{
|
||||
printf("Too low psnr = %gdb\n", psnr);
|
||||
imwrite("img.png", img);
|
||||
imwrite("img1.png", img1);
|
||||
ts->set_failed_test_info(ts->FAIL_MISMATCH);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
cvReleaseCapture( &cap );
|
||||
cvReleaseCapture( &saved );
|
||||
|
||||
ts->printf(ts->LOG, "end test function : ImagesVideo \n");
|
||||
@@ -306,7 +319,7 @@ void CV_HighGuiTest::SpecificImageTest(const string& dir)
|
||||
return;
|
||||
}
|
||||
|
||||
cv::resize(image, image, cv::Size(968, 757), 0.0, 0.0, cv::INTER_CUBIC);
|
||||
resize(image, image, Size(968, 757), 0.0, 0.0, INTER_CUBIC);
|
||||
|
||||
stringstream s_digit; s_digit << i;
|
||||
|
||||
@@ -375,247 +388,99 @@ void CV_HighGuiTest::SpecificImageTest(const string& dir)
|
||||
ts->set_failed_test_info(ts->OK);
|
||||
}
|
||||
|
||||
void CV_HighGuiTest::SpecificVideoFileTest(const string& dir, const char codecchars[4])
|
||||
{
|
||||
const string exts[] = {"avi", "mov", "mpg", "wmv"};
|
||||
const size_t n = sizeof(exts)/sizeof(exts[0]);
|
||||
int fourcc0 = CV_FOURCC(codecchars[0], codecchars[1], codecchars[2], codecchars[3]);
|
||||
|
||||
for (size_t j = 0; j < n; ++j)
|
||||
{
|
||||
string ext = exts[j];
|
||||
int fourcc = fourcc0;
|
||||
|
||||
if( (ext == "mov" && fourcc != CV_FOURCC('M', 'J', 'P', 'G')) ||
|
||||
(ext == "mpg" && fourcc != CV_FOURCC('m', 'p', 'e', 'g')) ||
|
||||
(ext == "wmv" && fourcc != CV_FOURCC('M', 'J', 'P', 'G')))
|
||||
continue;
|
||||
if( ext == "mov" )
|
||||
fourcc = CV_FOURCC('m', 'p', '4', 'v');
|
||||
|
||||
string fourcc_str = format("%c%c%c%c", fourcc & 255, (fourcc >> 8) & 255, (fourcc >> 16) & 255, (fourcc >> 24) & 255);
|
||||
const string video_file = "video_" + fourcc_str + "." + ext;
|
||||
|
||||
Size frame_size(968 & -2, 757 & -2);
|
||||
//Size frame_size(968 & -16, 757 & -16);
|
||||
//Size frame_size(640, 480);
|
||||
void CV_HighGuiTest::SpecificVideoTest(const string& dir, const cvtest::VideoFormat& fmt)
|
||||
{
|
||||
string ext = fmt.ext;
|
||||
int fourcc = fmt.fourcc;
|
||||
|
||||
string fourcc_str = cvtest::fourccToString(fourcc);
|
||||
const string video_file = "video_" + fourcc_str + "." + ext;
|
||||
|
||||
Size frame_size(968 & -2, 757 & -2);
|
||||
VideoWriter writer(video_file, fourcc, 25, frame_size, true);
|
||||
|
||||
if (!writer.isOpened())
|
||||
{
|
||||
// call it repeatedly for easier debugging
|
||||
VideoWriter writer(video_file, fourcc, 25, frame_size, true);
|
||||
|
||||
if (!writer.isOpened())
|
||||
{
|
||||
VideoWriter writer(video_file, fourcc, 25, frame_size, true);
|
||||
ts->printf(ts->LOG, "Creating a video in %s...\n", video_file.c_str());
|
||||
ts->printf(ts->LOG, "Cannot create VideoWriter object with codec %s.\n", fourcc_str.c_str());
|
||||
ts->set_failed_test_info(ts->FAIL_MISMATCH);
|
||||
continue;
|
||||
}
|
||||
|
||||
const size_t IMAGE_COUNT = 30;
|
||||
|
||||
for(size_t i = 0; i < IMAGE_COUNT; ++i)
|
||||
{
|
||||
stringstream s_digit;
|
||||
if (i < 10) {s_digit << "0"; s_digit << i;}
|
||||
else s_digit << i;
|
||||
|
||||
const string file_path = dir+"../python/images/QCIF_"+s_digit.str()+".bmp";
|
||||
|
||||
cv::Mat img = imread(file_path, CV_LOAD_IMAGE_COLOR);
|
||||
|
||||
if (img.empty())
|
||||
{
|
||||
ts->printf(ts->LOG, "Creating a video in %s...\n", video_file.c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot read frame from %s.\n", (ts->get_data_path()+"../python/images/QCIF_"+s_digit.str()+".bmp").c_str());
|
||||
ts->printf(ts->LOG, "Continue creating the video file...\n");
|
||||
ts->set_failed_test_info(ts->FAIL_INVALID_TEST_DATA);
|
||||
break;//continue;
|
||||
}
|
||||
|
||||
cv::resize(img, img, frame_size, 0.0, 0.0, cv::INTER_CUBIC);
|
||||
|
||||
for (int k = 0; k < img.rows; ++k)
|
||||
for (int l = 0; l < img.cols; ++l)
|
||||
if (img.at<Vec3b>(k, l) == Vec3b::all(0))
|
||||
img.at<Vec3b>(k, l) = Vec3b(0, 255, 0);
|
||||
else img.at<Vec3b>(k, l) = Vec3b(0, 0, 255);
|
||||
|
||||
imwrite("QCIF_"+s_digit.str()+".bmp", img);
|
||||
|
||||
writer << img;
|
||||
}
|
||||
|
||||
writer.release();
|
||||
cv::VideoCapture cap(video_file);
|
||||
|
||||
size_t FRAME_COUNT = (size_t)cap.get(CV_CAP_PROP_FRAME_COUNT);
|
||||
|
||||
if (FRAME_COUNT != IMAGE_COUNT && ext != "mpg" )
|
||||
{
|
||||
ts->printf(ts->LOG, "\nFrame count checking for video_%s.%s...\n", fourcc_str.c_str(), ext.c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", fourcc_str.c_str());
|
||||
ts->printf(ts->LOG, "Required frame count: %d; Returned frame count: %d\n", IMAGE_COUNT, FRAME_COUNT);
|
||||
ts->printf(ts->LOG, "Error: Incorrect frame count in the video.\n");
|
||||
ts->printf(ts->LOG, "Continue checking...\n");
|
||||
ts->set_failed_test_info(ts->FAIL_BAD_ACCURACY);
|
||||
}
|
||||
|
||||
//cap.set(CV_CAP_PROP_POS_FRAMES, -1);
|
||||
|
||||
for (int i = 0; i < (int)std::min<size_t>(FRAME_COUNT, IMAGE_COUNT)-1; i++)
|
||||
{
|
||||
cv::Mat frame; cap >> frame;
|
||||
if (frame.empty())
|
||||
{
|
||||
ts->printf(ts->LOG, "\nVideo file directory: %s\n", ".");
|
||||
ts->printf(ts->LOG, "File name: video_%s.%s\n", fourcc_str.c_str(), ext.c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", fourcc_str.c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot read the next frame with index %d.\n", i+1);
|
||||
ts->set_failed_test_info(ts->FAIL_MISSING_TEST_DATA);
|
||||
break;
|
||||
}
|
||||
|
||||
stringstream s_digit;
|
||||
if (i+1 < 10) {s_digit << "0"; s_digit << i+1;}
|
||||
else s_digit << i+1;
|
||||
|
||||
cv::Mat img = imread("QCIF_"+s_digit.str()+".bmp", CV_LOAD_IMAGE_COLOR);
|
||||
|
||||
if (img.empty())
|
||||
{
|
||||
ts->printf(ts->LOG, "\nError: cannot read an image from %s.\n", ("QCIF_"+s_digit.str()+".bmp").c_str());
|
||||
ts->set_failed_test_info(ts->FAIL_MISMATCH);
|
||||
continue;
|
||||
}
|
||||
|
||||
const double thresDbell = 40;
|
||||
|
||||
double psnr = PSNR(img, frame);
|
||||
|
||||
if (psnr > thresDbell)
|
||||
{
|
||||
ts->printf(ts->LOG, "\nReading frame from the file video_%s.%s...\n", fourcc_str.c_str(), ext.c_str());
|
||||
ts->printf(ts->LOG, "Frame index: %d\n", i+1);
|
||||
ts->printf(ts->LOG, "Difference between saved and original images: %g\n", psnr);
|
||||
ts->printf(ts->LOG, "Maximum allowed difference: %g\n", thresDbell);
|
||||
ts->printf(ts->LOG, "Error: too big difference between saved and original images.\n");
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void CV_HighGuiTest::SpecificVideoCameraTest(const string& dir, const char codecchars[4])
|
||||
{
|
||||
const string ext[] = {"avi", "mov", "mp4", "mpg", "wmv"};
|
||||
|
||||
const size_t n = sizeof(ext)/sizeof(ext[0]);
|
||||
|
||||
const int IMAGE_COUNT = 125;
|
||||
|
||||
cv::VideoCapture cap(0);
|
||||
|
||||
if (!cap.isOpened())
|
||||
{
|
||||
ts->printf(ts->LOG, "\nError: cannot start working with device.\n");
|
||||
ts->set_failed_test_info(ts->OK);
|
||||
ts->printf(ts->LOG, "Creating a video in %s...\n", video_file.c_str());
|
||||
ts->printf(ts->LOG, "Cannot create VideoWriter object with codec %s.\n", fourcc_str.c_str());
|
||||
ts->set_failed_test_info(ts->FAIL_MISMATCH);
|
||||
return;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < n; ++i)
|
||||
if ((ext[i]!="mp4")||(string(&codecchars[0], 4)!="IYUV"))
|
||||
#if defined WIN32 || defined _WIN32
|
||||
if (((ext[i]!="mov")||(string(&codecchars[0], 4)=="XVID"))&&(ext[i]!="mp4"))
|
||||
#endif
|
||||
const size_t IMAGE_COUNT = 30;
|
||||
vector<Mat> images;
|
||||
|
||||
for( size_t i = 0; i < IMAGE_COUNT; ++i )
|
||||
{
|
||||
Mat frame; int framecount = 0;
|
||||
cv::VideoWriter writer;
|
||||
string file_path = format("%s../python/images/QCIF_%02d.bmp", dir.c_str(), i);
|
||||
Mat img = imread(file_path, CV_LOAD_IMAGE_COLOR);
|
||||
|
||||
std::vector <cv::Mat> tmp_img(IMAGE_COUNT);
|
||||
if (img.empty())
|
||||
{
|
||||
ts->printf(ts->LOG, "Creating a video in %s...\n", video_file.c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot read frame from %s.\n", file_path.c_str());
|
||||
ts->printf(ts->LOG, "Continue creating the video file...\n");
|
||||
ts->set_failed_test_info(ts->FAIL_INVALID_TEST_DATA);
|
||||
break;
|
||||
}
|
||||
|
||||
writer.open("video_"+string(&codecchars[0], 4)+"."+ext[i], CV_FOURCC(codecchars[0], codecchars[1], codecchars[2], codecchars[3]), 25, Size(968, 757), true);
|
||||
for (int k = 0; k < img.rows; ++k)
|
||||
for (int l = 0; l < img.cols; ++l)
|
||||
if (img.at<Vec3b>(k, l) == Vec3b::all(0))
|
||||
img.at<Vec3b>(k, l) = Vec3b(0, 255, 0);
|
||||
else img.at<Vec3b>(k, l) = Vec3b(0, 0, 255);
|
||||
|
||||
resize(img, img, frame_size, 0.0, 0.0, INTER_CUBIC);
|
||||
|
||||
if (!writer.isOpened())
|
||||
images.push_back(img);
|
||||
writer << img;
|
||||
}
|
||||
|
||||
writer.release();
|
||||
VideoCapture cap(video_file);
|
||||
|
||||
size_t FRAME_COUNT = (size_t)cap.get(CV_CAP_PROP_FRAME_COUNT);
|
||||
|
||||
if (FRAME_COUNT != IMAGE_COUNT )
|
||||
{
|
||||
ts->printf(ts->LOG, "\nFrame count checking for video_%s.%s...\n", fourcc_str.c_str(), ext.c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", fourcc_str.c_str());
|
||||
ts->printf(ts->LOG, "Required frame count: %d; Returned frame count: %d\n", IMAGE_COUNT, FRAME_COUNT);
|
||||
ts->printf(ts->LOG, "Error: Incorrect frame count in the video.\n");
|
||||
ts->printf(ts->LOG, "Continue checking...\n");
|
||||
ts->set_failed_test_info(ts->FAIL_BAD_ACCURACY);
|
||||
return;
|
||||
}
|
||||
|
||||
for (int i = 0; (size_t)i < FRAME_COUNT; i++)
|
||||
{
|
||||
Mat frame; cap >> frame;
|
||||
if (frame.empty())
|
||||
{
|
||||
ts->printf(ts->LOG, "\nVideo file directory: %s\n", ".");
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", std::string(&codecchars[0], 4).c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot create VideoWriter object for video_%s.%s.\n", string(&codecchars[0]).c_str(), ext[i].c_str());
|
||||
ts->set_failed_test_info(ts->FAIL_EXCEPTION);
|
||||
continue;
|
||||
ts->printf(ts->LOG, "File name: video_%s.%s\n", fourcc_str.c_str(), ext.c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", fourcc_str.c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot read the next frame with index %d.\n", i+1);
|
||||
ts->set_failed_test_info(ts->FAIL_MISSING_TEST_DATA);
|
||||
break;
|
||||
}
|
||||
|
||||
for (;;)
|
||||
Mat img = images[i];
|
||||
|
||||
const double thresDbell = 40;
|
||||
double psnr = PSNR(img, frame);
|
||||
|
||||
if (psnr > thresDbell)
|
||||
{
|
||||
cap >> frame;
|
||||
|
||||
if (frame.empty())
|
||||
{
|
||||
ts->printf(ts->LOG, "\nVideo file directory: %s\n", ".");
|
||||
ts->printf(ts->LOG, "File name: video_%s.%s\n", string(&codecchars[0], 4).c_str(), ext[i].c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", string(&codecchars[0], 4).c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot read next frame with index %d from the device.\n", framecount);
|
||||
break;
|
||||
}
|
||||
|
||||
cv::resize(frame, frame, Size(968, 757), 0, 0, INTER_CUBIC);
|
||||
writer << frame; tmp_img[framecount] = frame;
|
||||
|
||||
framecount++;
|
||||
if (framecount == IMAGE_COUNT) break;
|
||||
}
|
||||
|
||||
cv::VideoCapture vcap(dir+"video_"+string(&codecchars[0], 4)+"."+ext[i]);
|
||||
|
||||
if (!vcap.isOpened())
|
||||
{
|
||||
ts->printf(ts->LOG, "\nVideo file directory: %s\n", ".");
|
||||
ts->printf(ts->LOG, "File name: video_%s.%s\n", string(&codecchars[0], 4).c_str(), ext[i].c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", string(&codecchars[0], 4).c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot open video file.\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
int FRAME_COUNT = (int)vcap.get(CV_CAP_PROP_FRAME_COUNT);
|
||||
|
||||
if (FRAME_COUNT != IMAGE_COUNT)
|
||||
{
|
||||
ts->printf(ts->LOG, "\nChecking frame count...\n");
|
||||
ts->printf(ts->LOG, "Video file directory: %s\n", ".");
|
||||
ts->printf(ts->LOG, "File name: video_%s.%s\n", string(&codecchars[0], 4).c_str(), ext[i].c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", string(&codecchars[0], 4).c_str());
|
||||
ts->printf(ts->LOG, "Required frame count: %d Returned frame count: %d\n", IMAGE_COUNT, FRAME_COUNT);
|
||||
ts->printf(ts->LOG, "Error: required and returned frame count are not matched.\n");
|
||||
ts->printf(ts->LOG, "Continue checking...\n");
|
||||
ts->set_failed_test_info(ts->FAIL_INVALID_OUTPUT);
|
||||
}
|
||||
|
||||
cv::Mat img; framecount = 0;
|
||||
vcap.set(CV_CAP_PROP_POS_FRAMES, 0);
|
||||
|
||||
for ( ; framecount < std::min<int>(FRAME_COUNT, IMAGE_COUNT); framecount++ )
|
||||
{
|
||||
vcap >> img;
|
||||
|
||||
if (img.empty())
|
||||
{
|
||||
ts->printf(ts->LOG, "\nVideo file directory: %s\n", ".");
|
||||
ts->printf(ts->LOG, "File name: video_%s.%s\n", string(&codecchars[0], 4).c_str(), ext[i].c_str());
|
||||
ts->printf(ts->LOG, "Video codec: %s\n", string(&codecchars[0], 4).c_str());
|
||||
ts->printf(ts->LOG, "Error: cannot read frame with index %d from the video.\n", framecount);
|
||||
break;
|
||||
}
|
||||
|
||||
const double thresDbell = 20;
|
||||
double psnr = PSNR(img, tmp_img[framecount]);
|
||||
|
||||
if (psnr > thresDbell)
|
||||
{
|
||||
ts->printf(ts->LOG, "\nReading frame from the file video_%s.%s...\n", string(&codecchars[0], 4).c_str(), ext[i].c_str());
|
||||
ts->printf(ts->LOG, "Frame index: %d\n", framecount);
|
||||
ts->printf(ts->LOG, "Difference between saved and original images: %g\n", psnr);
|
||||
ts->printf(ts->LOG, "Maximum allowed difference: %g\n", thresDbell);
|
||||
ts->printf(ts->LOG, "Error: too big difference between saved and original images.\n");
|
||||
continue;
|
||||
}
|
||||
ts->printf(ts->LOG, "\nReading frame from the file video_%s.%s...\n", fourcc_str.c_str(), ext.c_str());
|
||||
ts->printf(ts->LOG, "Frame index: %d\n", i+1);
|
||||
ts->printf(ts->LOG, "Difference between saved and original images: %g\n", psnr);
|
||||
ts->printf(ts->LOG, "Maximum allowed difference: %g\n", thresDbell);
|
||||
ts->printf(ts->LOG, "Error: too big difference between saved and original images.\n");
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -632,71 +497,33 @@ void CV_SpecificImageTest::run(int)
|
||||
|
||||
void CV_VideoTest::run(int)
|
||||
{
|
||||
#if defined WIN32 || (defined __linux__ && !defined ANDROID) || (defined __APPLE__ && defined HAVE_FFMPEG)
|
||||
#if !defined HAVE_GSTREAMER || defined HAVE_GSTREAMER_APP
|
||||
|
||||
const char codecs[][4] = { {'I', 'Y', 'U', 'V'},
|
||||
{'X', 'V', 'I', 'D'},
|
||||
{'m', 'p', 'e', 'g'},
|
||||
{'M', 'J', 'P', 'G'} };
|
||||
|
||||
printf("%s", ts->get_data_path().c_str());
|
||||
|
||||
int count = sizeof(codecs)/(4*sizeof(char));
|
||||
|
||||
for (int i = 0; i < count; ++i)
|
||||
for (int i = 0; ; ++i)
|
||||
{
|
||||
VideoTest(ts->get_data_path(), CV_FOURCC(codecs[i][0], codecs[i][1], codecs[i][2], codecs[i][3]));
|
||||
const cvtest::VideoFormat& fmt = cvtest::g_specific_fmt_list[i];
|
||||
if( fmt.empty() )
|
||||
break;
|
||||
VideoTest(ts->get_data_path(), fmt);
|
||||
}
|
||||
|
||||
#endif
|
||||
#endif
|
||||
}
|
||||
|
||||
void CV_SpecificVideoFileTest::run(int)
|
||||
void CV_SpecificVideoTest::run(int)
|
||||
{
|
||||
#if defined WIN32 || (defined __linux__ && !defined ANDROID) || (defined __APPLE__ && defined HAVE_FFMPEG)
|
||||
#if !defined HAVE_GSTREAMER || defined HAVE_GSTREAMER_APP
|
||||
|
||||
const char codecs[][4] = { {'m', 'p', 'e', 'g'},
|
||||
{'X', 'V', 'I', 'D'},
|
||||
{'M', 'J', 'P', 'G'},
|
||||
{'I', 'Y', 'U', 'V'} };
|
||||
|
||||
int count = sizeof(codecs)/(4*sizeof(char));
|
||||
|
||||
for (int i = 0; i < count; ++i)
|
||||
for (int i = 0; ; ++i)
|
||||
{
|
||||
SpecificVideoFileTest(ts->get_data_path(), codecs[i]);
|
||||
const cvtest::VideoFormat& fmt = cvtest::g_specific_fmt_list[i];
|
||||
if( fmt.empty() )
|
||||
break;
|
||||
SpecificVideoTest(ts->get_data_path(), fmt);
|
||||
}
|
||||
|
||||
#endif
|
||||
#endif
|
||||
}
|
||||
|
||||
void CV_SpecificVideoCameraTest::run(int)
|
||||
{
|
||||
#if defined WIN32 || (defined __linux__ && !defined ANDROID)
|
||||
#if !defined HAVE_GSTREAMER || defined HAVE_GSTREAMER_APP
|
||||
|
||||
const char codecs[][4] = { {'m', 'p', 'e', 'g'},
|
||||
{'X', 'V', 'I', 'D'},
|
||||
{'M', 'J', 'P', 'G'},
|
||||
{'I', 'Y', 'U', 'V'} };
|
||||
|
||||
int count = sizeof(codecs)/(4*sizeof(char));
|
||||
|
||||
for (int i = 0; i < count; ++i)
|
||||
{
|
||||
SpecificVideoCameraTest(ts->get_data_path(), codecs[i]);
|
||||
}
|
||||
|
||||
#endif
|
||||
#endif
|
||||
}
|
||||
|
||||
#ifdef HAVE_JPEG
|
||||
TEST(Highgui_Image, regression) { CV_ImageTest test; test.safe_run(); }
|
||||
#endif
|
||||
|
||||
#if BUILD_WITH_VIDEO_INPUT_SUPPORT && BUILD_WITH_VIDEO_OUTPUT_SUPPORT
|
||||
TEST(Highgui_Video, regression) { CV_VideoTest test; test.safe_run(); }
|
||||
TEST(Highgui_SpecificImage, regression) { CV_SpecificImageTest test; test.safe_run(); }
|
||||
TEST(Highgui_SpecificVideoFile, regression) { CV_SpecificVideoFileTest test; test.safe_run(); }
|
||||
TEST(Highgui_SpecificVideoCamera, regression) { CV_SpecificVideoCameraTest test; test.safe_run(); }
|
||||
TEST(Highgui_Video, write_read) { CV_SpecificVideoTest test; test.safe_run(); }
|
||||
#endif
|
||||
|
||||
TEST(Highgui_Image, write_read) { CV_SpecificImageTest test; test.safe_run(); }
|
||||
|
Reference in New Issue
Block a user