@@ -435,16 +435,16 @@ void GlobalMatting::expansionOfKnownRegions(cv::InputArray _img, cv::InputOutput
435
435
436
436
if (img.empty ())
437
437
CV_Error (CV_StsBadArg, " image is empty" );
438
- if (img. type () != CV_8UC3)
439
- CV_Error (CV_StsBadArg, " image mush have CV_8UC3 type" );
438
+
439
+ CV_CheckTypeEQ (img. type (),CV_8UC3, " image must have CV_8UC3 type" );
440
440
441
441
if (trimap.empty ())
442
442
CV_Error (CV_StsBadArg, " trimap is empty" );
443
- if (trimap. type () != CV_8UC1)
444
- CV_Error (CV_StsBadArg, " trimap mush have CV_8UC1 type" );
443
+
444
+ CV_CheckTypeEQ (trimap. type (),CV_8UC1, " trimap must have CV_8UC1 type" );
445
445
446
446
if (img.size () != trimap.size ())
447
- CV_Error (CV_StsBadArg, " image and trimap mush have same size" );
447
+ CV_Error (CV_StsBadArg, " image and trimap must have same size" );
448
448
449
449
for (int i = 0 ; i < niter; ++i)
450
450
expansionOfKnownRegionsHelper (img, trimap, i + 1 , float (niter - i));
@@ -520,13 +520,13 @@ void GlobalMatting::globalMatting(cv::InputArray _image, cv::InputArray _trimap,
520
520
521
521
if (image.empty ())
522
522
CV_Error (CV_StsBadArg, " image is empty" );
523
- if (image. type () != CV_8UC3)
524
- CV_Error (CV_StsBadArg, " image mush have CV_8UC3 type" );
523
+
524
+ CV_CheckTypeEQ (image,CV_8UC3, " image mush have CV_8UC3 type" );
525
525
526
526
if (trimap.empty ())
527
527
CV_Error (CV_StsBadArg, " trimap is empty" );
528
- if (trimap. type () != CV_8UC1)
529
- CV_Error (CV_StsBadArg, " trimap mush have CV_8UC1 type" );
528
+
529
+ CV_CheckTypeEQ (trimap,CV_8UC1, " trimap mush have CV_8UC1 type" );
530
530
531
531
if (image.size () != trimap.size ())
532
532
CV_Error (CV_StsBadArg, " image and trimap mush have same size" );
0 commit comments