search

bubbliiiing / retinanet-pytorch

这是一个retinanet-pytorch的源码，可以用于训练自己的模型。
MIT License
171 stars 27 forks source link

计算fps值报错 #12

Open Vegetabhl opened 2 years ago

Vegetabhl commented 2 years ago

您好，我用您的代码训练完，也测得了对应的map指标。但是在计算fps值的时候报了如下错误 TypeError: non_max_suppression() got multiple values for argument 'conf_thres'，麻烦请问下您有遇到过这个错误吗？ 我的环境是ubuntu下torch1.8，谢谢了。

bubbliiiing commented 2 years ago

多传了个参数，已修改