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

Update get_dr_txt.py

上级 6d08ca52
......@@ -59,8 +59,8 @@ class mAP_YOLO(YOLO):
self.input_image_shape = K.placeholder(shape=(2, ))
boxes, scores, classes = yolo_eval(self.yolo_model.output, self.anchors,
num_classes, self.input_image_shape,
score_threshold=self.score, iou_threshold=self.iou)
num_classes, self.input_image_shape, max_boxes = self.max_boxes,
score_threshold = self.score, iou_threshold = self.iou)
return boxes, scores, classes
#---------------------------------------------------#
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册