|
|
|
@ -469,7 +469,7 @@ size_t getDetectionIndices(
|
|
|
|
|
const size_t numClasses,
|
|
|
|
|
const size_t backgroundId,
|
|
|
|
|
const size_t batchSize,
|
|
|
|
|
const size_t confThreshold,
|
|
|
|
|
const real confThreshold,
|
|
|
|
|
const size_t nmsTopK,
|
|
|
|
|
const real nmsThreshold,
|
|
|
|
|
const size_t keepTopK,
|
|
|
|
@ -536,6 +536,8 @@ void getDetectionOutput(const real* confData,
|
|
|
|
|
MatrixPtr outBuffer;
|
|
|
|
|
Matrix::resizeOrCreate(outBuffer, numKept, 7, false, false);
|
|
|
|
|
real* bufferData = outBuffer->getData();
|
|
|
|
|
for (size_t i = 0; i < 7; i++)
|
|
|
|
|
bufferData[i] = -1;
|
|
|
|
|
size_t count = 0;
|
|
|
|
|
for (size_t n = 0; n < batchSize; ++n) {
|
|
|
|
|
for (map<size_t, vector<size_t>>::const_iterator it = allIndices[n].begin();
|
|
|
|
|