search

FateScript / CenterNet-better

An easy to understand and better performance version of CenterNet
Apache License 2.0
548 stars 104 forks source link

请问一下网络的后处理在哪个部分 #22

Closed GYxiaOH closed 4 years ago

GYxiaOH commented 4 years ago

rt。。。我看了下test里直接把model结果输出了？不是应该有个maxpool操作加上topk排序吗....我没有找到这个方面的实现？是有什么考虑吗？

FateScript commented 4 years ago

here

GYxiaOH commented 4 years ago

@FateScript 好的谢谢，您是在哪里调用这个函数的呢？我昨天追踪到here这里，就看到直接网络输出了。。。

FateScript commented 4 years ago

在这里，请善用github的检索功能，谢谢

GYxiaOH commented 4 years ago

啊，你误解我的意思了....不过我知道了 我没看到 if not self.training: return self.inference(images)这句话，现在逻辑清楚了！感谢您的帮助！