minor fix
This commit is contained in:
parent
da3a60a5a7
commit
631d4b483a
@ -742,7 +742,7 @@ namespace cv { namespace gpu { namespace bf_knnmatch
|
|||||||
{
|
{
|
||||||
matchUnrolledCached<16, 128, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
matchUnrolledCached<16, 128, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
}
|
}
|
||||||
else if (query.cols <= 256)
|
/*else if (query.cols <= 256)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 256, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
matchUnrolled<16, 256, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
}
|
}
|
||||||
@ -753,7 +753,7 @@ namespace cv { namespace gpu { namespace bf_knnmatch
|
|||||||
else if (query.cols <= 1024)
|
else if (query.cols <= 1024)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 1024, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
matchUnrolled<16, 1024, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
}
|
}*/
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
match<16, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
match<16, Dist>(query, train, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
@ -773,7 +773,7 @@ namespace cv { namespace gpu { namespace bf_knnmatch
|
|||||||
{
|
{
|
||||||
matchUnrolledCached<16, 128, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
matchUnrolledCached<16, 128, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
}
|
}
|
||||||
else if (query.cols <= 256)
|
/*else if (query.cols <= 256)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 256, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
matchUnrolled<16, 256, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
}
|
}
|
||||||
@ -784,7 +784,7 @@ namespace cv { namespace gpu { namespace bf_knnmatch
|
|||||||
else if (query.cols <= 1024)
|
else if (query.cols <= 1024)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 1024, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
matchUnrolled<16, 1024, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
}
|
}*/
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
match<16, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
match<16, Dist>(query, trains, n, mask, static_cast< DevMem2D_<int2> >(trainIdx), static_cast< DevMem2D_<int2> >(imgIdx), static_cast< DevMem2D_<float2> > (distance), stream);
|
||||||
@ -937,7 +937,7 @@ namespace cv { namespace gpu { namespace bf_knnmatch
|
|||||||
{
|
{
|
||||||
calcDistanceUnrolled<16, 128, Dist>(query, train, mask, allDist, stream);
|
calcDistanceUnrolled<16, 128, Dist>(query, train, mask, allDist, stream);
|
||||||
}
|
}
|
||||||
else if (query.cols <= 256)
|
/*else if (query.cols <= 256)
|
||||||
{
|
{
|
||||||
calcDistanceUnrolled<16, 256, Dist>(query, train, mask, allDist, stream);
|
calcDistanceUnrolled<16, 256, Dist>(query, train, mask, allDist, stream);
|
||||||
}
|
}
|
||||||
@ -948,7 +948,7 @@ namespace cv { namespace gpu { namespace bf_knnmatch
|
|||||||
else if (query.cols <= 1024)
|
else if (query.cols <= 1024)
|
||||||
{
|
{
|
||||||
calcDistanceUnrolled<16, 1024, Dist>(query, train, mask, allDist, stream);
|
calcDistanceUnrolled<16, 1024, Dist>(query, train, mask, allDist, stream);
|
||||||
}
|
}*/
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
calcDistance<16, Dist>(query, train, mask, allDist, stream);
|
calcDistance<16, Dist>(query, train, mask, allDist, stream);
|
||||||
|
@ -536,7 +536,7 @@ namespace cv { namespace gpu { namespace bf_match
|
|||||||
const dim3 block(BLOCK_SIZE, BLOCK_SIZE);
|
const dim3 block(BLOCK_SIZE, BLOCK_SIZE);
|
||||||
const dim3 grid(divUp(query.rows, BLOCK_SIZE));
|
const dim3 grid(divUp(query.rows, BLOCK_SIZE));
|
||||||
|
|
||||||
const size_t smemSize = (2 * BLOCK_SIZE * BLOCK_SIZE) * sizeof(int);
|
const size_t smemSize = (3 * BLOCK_SIZE * BLOCK_SIZE) * sizeof(int);
|
||||||
|
|
||||||
match<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, trains, n, mask, trainIdx.data, imgIdx.data, distance.data);
|
match<BLOCK_SIZE, Dist><<<grid, block, smemSize, stream>>>(query, trains, n, mask, trainIdx.data, imgIdx.data, distance.data);
|
||||||
cudaSafeCall( cudaGetLastError() );
|
cudaSafeCall( cudaGetLastError() );
|
||||||
@ -561,7 +561,7 @@ namespace cv { namespace gpu { namespace bf_match
|
|||||||
{
|
{
|
||||||
matchUnrolledCached<16, 128, Dist>(query, train, mask, trainIdx, distance, stream);
|
matchUnrolledCached<16, 128, Dist>(query, train, mask, trainIdx, distance, stream);
|
||||||
}
|
}
|
||||||
else if (query.cols <= 256)
|
/*else if (query.cols <= 256)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 256, Dist>(query, train, mask, trainIdx, distance, stream);
|
matchUnrolled<16, 256, Dist>(query, train, mask, trainIdx, distance, stream);
|
||||||
}
|
}
|
||||||
@ -572,7 +572,7 @@ namespace cv { namespace gpu { namespace bf_match
|
|||||||
else if (query.cols <= 1024)
|
else if (query.cols <= 1024)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 1024, Dist>(query, train, mask, trainIdx, distance, stream);
|
matchUnrolled<16, 1024, Dist>(query, train, mask, trainIdx, distance, stream);
|
||||||
}
|
}*/
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
match<16, Dist>(query, train, mask, trainIdx, distance, stream);
|
match<16, Dist>(query, train, mask, trainIdx, distance, stream);
|
||||||
@ -592,7 +592,7 @@ namespace cv { namespace gpu { namespace bf_match
|
|||||||
{
|
{
|
||||||
matchUnrolledCached<16, 128, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
matchUnrolledCached<16, 128, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
||||||
}
|
}
|
||||||
else if (query.cols <= 256)
|
/*else if (query.cols <= 256)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 256, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
matchUnrolled<16, 256, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
||||||
}
|
}
|
||||||
@ -603,7 +603,7 @@ namespace cv { namespace gpu { namespace bf_match
|
|||||||
else if (query.cols <= 1024)
|
else if (query.cols <= 1024)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 1024, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
matchUnrolled<16, 1024, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
||||||
}
|
}*/
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
match<16, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
match<16, Dist>(query, trains, n, mask, trainIdx, imgIdx, distance, stream);
|
||||||
|
@ -283,7 +283,7 @@ namespace cv { namespace gpu { namespace bf_radius_match
|
|||||||
{
|
{
|
||||||
matchUnrolled<16, 128, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
matchUnrolled<16, 128, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
||||||
}
|
}
|
||||||
else if (query.cols <= 256)
|
/*else if (query.cols <= 256)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 256, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
matchUnrolled<16, 256, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
||||||
}
|
}
|
||||||
@ -294,7 +294,7 @@ namespace cv { namespace gpu { namespace bf_radius_match
|
|||||||
else if (query.cols <= 1024)
|
else if (query.cols <= 1024)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 1024, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
matchUnrolled<16, 1024, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
||||||
}
|
}*/
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
match<16, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
match<16, Dist>(query, train, maxDistance, mask, trainIdx, distance, nMatches, stream);
|
||||||
@ -314,7 +314,7 @@ namespace cv { namespace gpu { namespace bf_radius_match
|
|||||||
{
|
{
|
||||||
matchUnrolled<16, 128, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
matchUnrolled<16, 128, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
||||||
}
|
}
|
||||||
else if (query.cols <= 256)
|
/*else if (query.cols <= 256)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 256, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
matchUnrolled<16, 256, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
||||||
}
|
}
|
||||||
@ -325,7 +325,7 @@ namespace cv { namespace gpu { namespace bf_radius_match
|
|||||||
else if (query.cols <= 1024)
|
else if (query.cols <= 1024)
|
||||||
{
|
{
|
||||||
matchUnrolled<16, 1024, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
matchUnrolled<16, 1024, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
||||||
}
|
}*/
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
match<16, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
match<16, Dist>(query, trains, n, maxDistance, masks, trainIdx, imgIdx, distance, nMatches, stream);
|
||||||
|
Loading…
x
Reference in New Issue
Block a user