fixed getUMatIndex
This commit is contained in:
parent
a7d2830d3f
commit
52b8bb6761
@ -1940,10 +1940,16 @@ static void getUMatIndex(const std::vector<UMat> & um, int cn, int & idx, int &
|
|||||||
int ccn = um[i].channels();
|
int ccn = um[i].channels();
|
||||||
totalChannels += ccn;
|
totalChannels += ccn;
|
||||||
|
|
||||||
if (totalChannels >= cn)
|
if (totalChannels == cn)
|
||||||
|
{
|
||||||
|
idx = (int)(i + 1);
|
||||||
|
cnidx = 0;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if (totalChannels > cn)
|
||||||
{
|
{
|
||||||
idx = (int)i;
|
idx = (int)i;
|
||||||
cnidx = i == 0 ? cn : cn % (totalChannels - ccn);
|
cnidx = i == 0 ? cn : (cn - totalChannels + ccn);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user