未验证 提交 2284acb3 编写于 作者: B Bubbliiiing 提交者: GitHub

Update get_dr_txt.py

上级 789c2b6c
......@@ -54,7 +54,7 @@ class mAP_EfficientDet(EfficientDet):
detection = torch.cat([regression,classification],axis=-1)
batch_detections = non_max_suppression(detection, len(self.class_names),
conf_thres=self.confidence,
nms_thres=0.2)
nms_thres=0.3)
try:
batch_detections = batch_detections[0].cpu().numpy()
except:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册