mirror of
https://github.com/opencv/opencv.git
synced 2024-12-11 22:59:16 +08:00
87b569d812
general: - all iterative tests have been replaced with parameterized tests - old-style try..catch tests have been modified to use EXPECT_/ASSERT_ gtest macros - added temporary files cleanup - modified MatComparator error message formatting imgcodecs: - test_grfmt.cpp split to test_jpg.cpp, test_png.cpp, test_tiff.cpp, etc. videoio: - added public HAVE_VIDEO_INPUT, HAVE_VIDEO_OUTPUT definitions to cvconfig.h - built-in MotionJPEG codec could not be tested on some platforms (read_write test was disabled if ffmpeg is off, encoding/decoding was handled by ffmpeg otherwise). - image-related tests moved to imgcodecs (Videoio_Image) - several property get/set tests have been combined into one - added MotionJPEG test video to opencv_extra
146 lines
3.8 KiB
C++
146 lines
3.8 KiB
C++
#include "../test_precomp.hpp"
|
|
#include "opencv2/ts/ocl_test.hpp"
|
|
|
|
#ifdef HAVE_OPENCL
|
|
#ifdef HAVE_VIDEO_INPUT
|
|
|
|
namespace cvtest {
|
|
namespace ocl {
|
|
|
|
//////////////////////////Mog2_Update///////////////////////////////////
|
|
|
|
namespace
|
|
{
|
|
IMPLEMENT_PARAM_CLASS(UseGray, bool)
|
|
IMPLEMENT_PARAM_CLASS(DetectShadow, bool)
|
|
IMPLEMENT_PARAM_CLASS(UseFloat, bool)
|
|
}
|
|
|
|
PARAM_TEST_CASE(Mog2_Update, UseGray, DetectShadow,UseFloat)
|
|
{
|
|
bool useGray;
|
|
bool detectShadow;
|
|
bool useFloat;
|
|
virtual void SetUp()
|
|
{
|
|
useGray = GET_PARAM(0);
|
|
detectShadow = GET_PARAM(1);
|
|
useFloat = GET_PARAM(2);
|
|
}
|
|
};
|
|
|
|
OCL_TEST_P(Mog2_Update, Accuracy)
|
|
{
|
|
string inputFile = string(TS::ptr()->get_data_path()) + "video/768x576.avi";
|
|
VideoCapture cap(inputFile);
|
|
ASSERT_TRUE(cap.isOpened());
|
|
|
|
Ptr<BackgroundSubtractorMOG2> mog2_cpu = createBackgroundSubtractorMOG2();
|
|
Ptr<BackgroundSubtractorMOG2> mog2_ocl = createBackgroundSubtractorMOG2();
|
|
|
|
mog2_cpu->setDetectShadows(detectShadow);
|
|
mog2_ocl->setDetectShadows(detectShadow);
|
|
|
|
Mat frame, foreground;
|
|
UMat u_foreground;
|
|
|
|
for (int i = 0; i < 10; ++i)
|
|
{
|
|
cap >> frame;
|
|
ASSERT_FALSE(frame.empty());
|
|
|
|
if (useGray)
|
|
{
|
|
Mat temp;
|
|
cvtColor(frame, temp, COLOR_BGR2GRAY);
|
|
swap(temp, frame);
|
|
}
|
|
|
|
if(useFloat)
|
|
{
|
|
Mat temp;
|
|
frame.convertTo(temp,CV_32F);
|
|
swap(temp,frame);
|
|
}
|
|
|
|
OCL_OFF(mog2_cpu->apply(frame, foreground));
|
|
OCL_ON (mog2_ocl->apply(frame, u_foreground));
|
|
|
|
if (detectShadow)
|
|
EXPECT_MAT_SIMILAR(foreground, u_foreground, 15e-3);
|
|
else
|
|
EXPECT_MAT_NEAR(foreground, u_foreground, 0);
|
|
}
|
|
}
|
|
|
|
//////////////////////////Mog2_getBackgroundImage///////////////////////////////////
|
|
|
|
PARAM_TEST_CASE(Mog2_getBackgroundImage, DetectShadow, UseFloat)
|
|
{
|
|
bool detectShadow;
|
|
bool useFloat;
|
|
virtual void SetUp()
|
|
{
|
|
detectShadow = GET_PARAM(0);
|
|
useFloat = GET_PARAM(1);
|
|
}
|
|
};
|
|
|
|
OCL_TEST_P(Mog2_getBackgroundImage, Accuracy)
|
|
{
|
|
string inputFile = string(TS::ptr()->get_data_path()) + "video/768x576.avi";
|
|
VideoCapture cap(inputFile);
|
|
ASSERT_TRUE(cap.isOpened());
|
|
|
|
Ptr<BackgroundSubtractorMOG2> mog2_cpu = createBackgroundSubtractorMOG2();
|
|
Ptr<BackgroundSubtractorMOG2> mog2_ocl = createBackgroundSubtractorMOG2();
|
|
|
|
mog2_cpu->setDetectShadows(detectShadow);
|
|
mog2_ocl->setDetectShadows(detectShadow);
|
|
|
|
Mat frame, foreground;
|
|
UMat u_foreground;
|
|
|
|
for (int i = 0; i < 10; ++i)
|
|
{
|
|
cap >> frame;
|
|
ASSERT_FALSE(frame.empty());
|
|
|
|
if(useFloat)
|
|
{
|
|
Mat temp;
|
|
frame.convertTo(temp,CV_32F);
|
|
swap(temp,frame);
|
|
}
|
|
|
|
OCL_OFF(mog2_cpu->apply(frame, foreground));
|
|
OCL_ON (mog2_ocl->apply(frame, u_foreground));
|
|
}
|
|
|
|
Mat background;
|
|
OCL_OFF(mog2_cpu->getBackgroundImage(background));
|
|
|
|
UMat u_background;
|
|
OCL_ON (mog2_ocl->getBackgroundImage(u_background));
|
|
|
|
EXPECT_MAT_NEAR(background, u_background, 1.0);
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
OCL_INSTANTIATE_TEST_CASE_P(OCL_Video, Mog2_Update, Combine(
|
|
Values(UseGray(true),UseGray(false)),
|
|
Values(DetectShadow(true), DetectShadow(false)),
|
|
Values(UseFloat(false),UseFloat(true)))
|
|
);
|
|
|
|
OCL_INSTANTIATE_TEST_CASE_P(OCL_Video, Mog2_getBackgroundImage, Combine(
|
|
Values(DetectShadow(true), DetectShadow(false)),
|
|
Values(UseFloat(false),UseFloat(true)))
|
|
);
|
|
|
|
}}// namespace cvtest::ocl
|
|
|
|
#endif
|
|
#endif
|