TensorRT-CenterNet icon indicating copy to clipboard operation
TensorRT-CenterNet copied to clipboard

about the outputBufferSize

Open lyupei opened this issue 5 years ago • 0 comments

Hi, I wanna ask why the outputBufferSize is set as mBindBufferSizes[1] * 6 in line 141 of ctdetNet.cpp?

This may cause overflow(in extreme cases) because each detection result contain 15 float members and 1 int member.

lyupei avatar Aug 18 '20 08:08 lyupei