未验证 提交 3b28629e 编写于 作者: F FlyingQianMM 提交者: GitHub

fix dtype error in retinanet_target_assgin example codes. test=develop (#25091)

上级 9ab3cf03
...@@ -230,9 +230,9 @@ def retinanet_target_assign(bbox_pred, ...@@ -230,9 +230,9 @@ def retinanet_target_assign(bbox_pred,
gt_boxes = fluid.data(name='gt_boxes', shape=[10, 4], gt_boxes = fluid.data(name='gt_boxes', shape=[10, 4],
dtype='float32') dtype='float32')
gt_labels = fluid.data(name='gt_labels', shape=[10, 1], gt_labels = fluid.data(name='gt_labels', shape=[10, 1],
dtype='float32') dtype='int32')
is_crowd = fluid.data(name='is_crowd', shape=[1], is_crowd = fluid.data(name='is_crowd', shape=[1],
dtype='float32') dtype='int32')
im_info = fluid.data(name='im_info', shape=[1, 3], im_info = fluid.data(name='im_info', shape=[1, 3],
dtype='float32') 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 = \\
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册