mirror of
https://github.com/zebrajr/opencv.git
synced 2025-12-06 00:19:46 +01:00
0ee9c27966
2025-10-14T05:53:31.5387050Z C:\GHA-OCV-1\_work\ci-gha-workflow\ci-gha-workflow\opencv\modules\imgcodecs\src\bitstrm.cpp(156,57): warning C4244: 'argument': conversion from 'int64_t' to 'ptrdiff_t', possible loss of data [C:\GHA-OCV-1\_work\ci-gha-workflow\ci-gha-workflow\build\modules\imgcodecs\opencv_imgcodecs.vcxproj] ### Pull Request Readiness Checklist Optional Known Foreground Mask for Background Subtractors #27810 See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request - [x] I agree to contribute to the project under Apache 2 License. - [x] To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV - [x] The PR is proposed to the proper branch - [x] There is a reference to the original bug report and related work - [x] There is accuracy test, performance test and test data in opencv_extra repository, if applicable Patch to opencv_extra has the same branch name. - [x] The feature is well documented and sample code can be built with the project CMake ### Description This adds an optional foreground input mask parameter to the MOG2 and KNN background subtractors, in line with issue https://github.com/opencv/opencv/issues/26476 4 tests are added under test_bgfg2.cpp: 2 for each subtractor type (1 with shadow detection and 1 without) A demo shows the feature with only 3 parameters and with a 4th optional foreground mask for both core subtractor types. Note: To patch contrib inheritance of the background subtraction class, empty apply method which throws a not implemented error is added to contrib subclasses. This is done to keep the overloaded apply function as pure virtual. Contrib PR to be made and linked shortly. Contrib Repo Paired Pull Request: https://github.com/opencv/opencv_contrib/pull/4017
109 lines
2.9 KiB
C++
109 lines
2.9 KiB
C++
// 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 "test_precomp.hpp"
|
|
#include "opencv2/video/background_segm.hpp"
|
|
|
|
namespace opencv_test { namespace {
|
|
|
|
using namespace cv;
|
|
|
|
///////////////////////// MOG2 //////////////////////////////
|
|
TEST(BackgroundSubtractorMOG2, KnownForegroundMaskShadowsTrue)
|
|
{
|
|
Ptr<BackgroundSubtractorMOG2> mog2 = createBackgroundSubtractorMOG2(500, 16, true);
|
|
|
|
//Black Frame
|
|
Mat input = Mat::zeros(480,640 , CV_8UC3);
|
|
|
|
//White Rectangle
|
|
Mat knownFG = Mat::zeros(input.size(), CV_8U);
|
|
|
|
rectangle(knownFG, Rect(3,3,5,5), Scalar(255,255,255), -1);
|
|
|
|
Mat output;
|
|
mog2->apply(input, knownFG, output);
|
|
|
|
for(int y = 3; y < 8; y++)
|
|
{
|
|
for (int x = 3; x < 8; x++){
|
|
EXPECT_EQ(255,output.at<uchar>(y,x)) << "Expected foreground at (" << x << "," << y << ")";
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(BackgroundSubtractorMOG2, KnownForegroundMaskShadowsFalse)
|
|
{
|
|
Ptr<BackgroundSubtractorMOG2> mog2 = createBackgroundSubtractorMOG2(500, 16, false);
|
|
|
|
//Black Frame
|
|
Mat input = Mat::zeros(480,640 , CV_8UC3);
|
|
|
|
//White Rectangle
|
|
Mat knownFG = Mat::zeros(input.size(), CV_8U);
|
|
|
|
rectangle(knownFG, Rect(3,3,5,5), Scalar(255,255,255), FILLED);
|
|
|
|
Mat output;
|
|
mog2->apply(input, knownFG, output);
|
|
|
|
for(int y = 3; y < 8; y++)
|
|
{
|
|
for (int x = 3; x < 8; x++){
|
|
EXPECT_EQ(255,output.at<uchar>(y,x)) << "Expected foreground at (" << x << "," << y << ")";
|
|
}
|
|
}
|
|
}
|
|
|
|
///////////////////////// KNN //////////////////////////////
|
|
|
|
TEST(BackgroundSubtractorKNN, KnownForegroundMaskShadowsTrue)
|
|
{
|
|
Ptr<BackgroundSubtractorKNN> knn = createBackgroundSubtractorKNN(500, 400.0, true);
|
|
|
|
//Black Frame
|
|
Mat input = Mat::zeros(480,640 , CV_8UC3);
|
|
|
|
//White Rectangle
|
|
Mat knownFG = Mat::zeros(input.size(), CV_8U);
|
|
|
|
rectangle(knownFG, Rect(3,3,5,5), Scalar(255,255,255), FILLED);
|
|
|
|
Mat output;
|
|
knn->apply(input, knownFG, output);
|
|
|
|
for(int y = 3; y < 8; y++)
|
|
{
|
|
for (int x = 3; x < 8; x++){
|
|
EXPECT_EQ(255,output.at<uchar>(y,x)) << "Expected foreground at (" << x << "," << y << ")";
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(BackgroundSubtractorKNN, KnownForegroundMaskShadowsFalse)
|
|
{
|
|
Ptr<BackgroundSubtractorKNN> knn = createBackgroundSubtractorKNN(500, 400.0, false);
|
|
|
|
//Black Frame
|
|
Mat input = Mat::zeros(480,640 , CV_8UC3);
|
|
|
|
//White Rectangle
|
|
Mat knownFG = Mat::zeros(input.size(), CV_8U);
|
|
|
|
rectangle(knownFG, Rect(3,3,5,5), Scalar(255,255,255), FILLED);
|
|
|
|
Mat output;
|
|
knn->apply(input, knownFG, output);
|
|
|
|
for(int y = 3; y < 8; y++)
|
|
{
|
|
for (int x = 3; x < 8; x++){
|
|
EXPECT_EQ(255,output.at<uchar>(y,x)) << "Expected foreground at (" << x << "," << y << ")";
|
|
}
|
|
}
|
|
}
|
|
|
|
}} // namespace
|
|
/* End of file. */
|