diff --git a/python/paddle/fluid/layers/detection.py b/python/paddle/fluid/layers/detection.py index a4fa34e8fb3fc28e6f8485771065b1c72738d4ae..e0725242b7b28f92a9d788a17ec2fe84fcdacc83 100644 --- a/python/paddle/fluid/layers/detection.py +++ b/python/paddle/fluid/layers/detection.py @@ -233,7 +233,7 @@ def retinanet_target_assign(bbox_pred, dtype='float32') im_info = fluid.data(name='im_infoss', shape=[1, 3], dtype='float32') - score_pred, loc_pred, score_target, loc_target, bbox_inside_weight, fg_num = + score_pred, loc_pred, score_target, loc_target, bbox_inside_weight, fg_num = \\ fluid.layers.retinanet_target_assign(bbox_pred, cls_logits, anchor_box, anchor_var, gt_boxes, gt_labels, is_crowd, im_info, 10)