vendor: OpenCV 5.0.0 snapshot at 40738fb16ceddb5fb3fea747585f7ce6abb0605b
This commit is contained in:
@@ -0,0 +1,88 @@
|
||||
// This file is part of OpenCV project.
|
||||
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
||||
// of this distribution and at http://opencv.org/license.html.
|
||||
|
||||
#include "perf_precomp.hpp"
|
||||
|
||||
#include "perf_bgfg_utils.hpp"
|
||||
|
||||
namespace opencv_test { namespace {
|
||||
|
||||
//////////////////////////// KNN//////////////////////////
|
||||
|
||||
typedef tuple<std::string, int> VideoKNNParamType;
|
||||
typedef TestBaseWithParam<VideoKNNParamType> KNN_Apply;
|
||||
typedef TestBaseWithParam<VideoKNNParamType> KNN_GetBackgroundImage;
|
||||
|
||||
PERF_TEST_P(KNN_Apply, KNN, Combine(Values("cv/video/768x576.avi", "cv/video/1920x1080.avi"), Values(1,3)))
|
||||
{
|
||||
VideoKNNParamType params = GetParam();
|
||||
|
||||
const string inputFile = getDataPath(get<0>(params));
|
||||
|
||||
const int cn = get<1>(params);
|
||||
int nFrame = 5;
|
||||
|
||||
vector<Mat> frame_buffer(nFrame);
|
||||
|
||||
cv::VideoCapture cap(inputFile);
|
||||
if (!cap.isOpened())
|
||||
throw SkipTestException("Video file can not be opened");
|
||||
prepareData(cap, cn, frame_buffer);
|
||||
|
||||
Mat foreground;
|
||||
|
||||
TEST_CYCLE()
|
||||
{
|
||||
Ptr<cv::BackgroundSubtractorKNN> knn = createBackgroundSubtractorKNN();
|
||||
knn->setDetectShadows(false);
|
||||
foreground.release();
|
||||
for (int i = 0; i < nFrame; i++)
|
||||
{
|
||||
knn->apply(frame_buffer[i], foreground);
|
||||
}
|
||||
}
|
||||
SANITY_CHECK_NOTHING();
|
||||
}
|
||||
|
||||
PERF_TEST_P(KNN_GetBackgroundImage, KNN, Values(
|
||||
std::make_pair<string, int>("cv/video/768x576.avi", 5),
|
||||
std::make_pair<string, int>("cv/video/1920x1080.avi", 5)))
|
||||
{
|
||||
VideoKNNParamType params = GetParam();
|
||||
|
||||
const string inputFile = getDataPath(get<0>(params));
|
||||
|
||||
const int cn = 3;
|
||||
const int skipFrames = get<1>(params);
|
||||
int nFrame = 10;
|
||||
|
||||
vector<Mat> frame_buffer(nFrame);
|
||||
|
||||
cv::VideoCapture cap(inputFile);
|
||||
if (!cap.isOpened())
|
||||
throw SkipTestException("Video file can not be opened");
|
||||
prepareData(cap, cn, frame_buffer, skipFrames);
|
||||
|
||||
Mat foreground, background;
|
||||
|
||||
TEST_CYCLE()
|
||||
{
|
||||
Ptr<cv::BackgroundSubtractorKNN> knn = createBackgroundSubtractorKNN();
|
||||
knn->setDetectShadows(false);
|
||||
foreground.release();
|
||||
background.release();
|
||||
for (int i = 0; i < nFrame; i++)
|
||||
{
|
||||
knn->apply(frame_buffer[i], foreground);
|
||||
}
|
||||
knn->getBackgroundImage(background);
|
||||
}
|
||||
#ifdef DEBUG_BGFG
|
||||
imwrite(format("fg_%d_%d_knn.png", frame_buffer[0].rows, cn), foreground);
|
||||
imwrite(format("bg_%d_%d_knn.png", frame_buffer[0].rows, cn), background);
|
||||
#endif
|
||||
SANITY_CHECK_NOTHING();
|
||||
}
|
||||
|
||||
}}// namespace
|
||||
Reference in New Issue
Block a user