search

Ghustwb / MobileNet-SSD-TensorRT

Accelerate mobileNet-ssd with tensorRT
188 stars 91 forks source link

关于检测结果抖动的问题 #42

Open JullyZ opened 4 years ago

JullyZ commented 4 years ago

hi，你好，我从你的博客过来，我看到你的博客里说解决了检测抖动的问题，是softmax层的问题，可以请问一下是softmax哪里实现的问题吗？谢谢。

EvaneSnow commented 4 years ago

@JullyZ 你好，请问你解决了检测抖动问题了吗

JullyZ commented 4 years ago

@JullyZ 你好，请问你解决了检测抖动问题了吗

https://github.com/Ghustwb/MobileNet-SSD-TensorRT/blob/6e55d211462fb6de52f8fabbae0287e358983914/kernel.cu#L180 这里。

EvaneSnow commented 4 years ago

@JullyZ 你可以告诉我怎么修改吗？我有点不会，谢谢