Update seamless_cloning.cpp

5 years ago · eff514f8bb
parent 955b20230c
commit eff514f8bb
1 changed files with 2 additions and 2 deletions
--- a/modules/photo/src/seamless_cloning.cpp
+++ b/modules/photo/src/seamless_cloning.cpp
@ -51,8 +51,8 @@ static Mat checkMask(InputArray _mask, Size size)
 {
    Mat mask = _mask.getMat();
    Mat gray;
-    if (mask.channels() == 3)
-        cvtColor(mask, gray, COLOR_BGR2GRAY);
+    if (mask.channels() > 1)
+        cvtColor(mask, gray, COLOR_BGRA2GRAY);
    else
    {
        if (mask.empty())