search

bubbliiiing / efficientdet-pytorch

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

用predict.py计算fps时出现的问题 #32

Open sfx-666 opened 2 years ago

sfx-666 commented 2 years ago

Traceback (most recent call last): File "e:/deeplearning/efficientdet-pytorch-master/efficientdet-pytorch-master/predict.py", line 120, in tact_time = efficientdet.get_FPS(img, test_interval) File "e:\deeplearning\efficientdet-pytorch-master\efficientdet-pytorch-master\efficientdet.py", line 237, in get_FPS image_shape, self.letterbox_image, conf_thres = self.confidence, nms_thres = self.nms_iou) TypeError: non_max_suppression() got multiple values for argument 'conf_thres'

bubbliiiing commented 2 years ago

修改了