Update seamless_cloning.cpp

2019-11-17 21:25:03 +03:00
parent 955b20230c
commit eff514f8bb
1 changed files with 2 additions and 2 deletions
@@ -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())