diff --git a/PaddleCV/yolov3/models/yolov3.py b/PaddleCV/yolov3/models/yolov3.py index 55c0667b16097b4424c948798a5faccc8ad2e366..873968337b18badae020653577efc2ec230573a1 100644 --- a/PaddleCV/yolov3/models/yolov3.py +++ b/PaddleCV/yolov3/models/yolov3.py @@ -139,9 +139,9 @@ class YOLOv3(object): if self.is_train: loss = fluid.layers.yolov3_loss( x=out, - gtbox=self.gtbox, - gtlabel=self.gtlabel, - gtscore=self.gtscore, + gt_box=self.gtbox, + gt_label=self.gtlabel, + gt_score=self.gtscore, anchors=cfg.anchors, anchor_mask=anchor_mask, class_num=cfg.class_num,