retinanet-pytorch icon indicating copy to clipboard operation
retinanet-pytorch copied to clipboard

计算fps值报错

Open Vegetabhl opened this issue 4 years ago • 1 comments

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

Vegetabhl avatar Nov 29 '21 13:11 Vegetabhl

多传了个参数,已修改

bubbliiiing avatar Nov 29 '21 14:11 bubbliiiing