Skip to content

Commit 6768e48

Browse files
committed
Merge pull request #651 from cbalint13:seeds
2 parents 1043e2e + 6241296 commit 6768e48

File tree

1 file changed

+25
-1
lines changed

1 file changed

+25
-1
lines changed

modules/ximgproc/src/seeds.cpp

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -375,7 +375,31 @@ void SuperpixelSEEDSImpl::initImageBins<float>(const Mat& img, int)
375375

376376
void SuperpixelSEEDSImpl::initImage(InputArray img)
377377
{
378-
Mat src = img.getMat();
378+
Mat src;
379+
380+
if ( img.isMat() )
381+
{
382+
// get Mat
383+
src = img.getMat();
384+
385+
// image should be valid
386+
CV_Assert( !src.empty() );
387+
}
388+
else if ( img.isMatVector() )
389+
{
390+
vector<Mat> vec;
391+
// get vector Mat
392+
img.getMatVector( vec );
393+
394+
// array should be valid
395+
CV_Assert( !vec.empty() );
396+
397+
// merge into Mat
398+
merge( vec, src );
399+
}
400+
else
401+
CV_Error( Error::StsInternal, "Invalid InputArray." );
402+
379403
int depth = src.depth();
380404
seeds_current_level = seeds_nr_levels - 2;
381405
forwardbackward = true;

0 commit comments

Comments
 (0)