search

CaoWGG / TensorRT-CenterNet

tensorrt5 , centernet , centerface, deform conv, int8
767 stars 159 forks source link

about the outputBufferSize #78

Open lyupei opened 4 years ago

lyupei commented 4 years ago

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.