Skip to content

P
PaddleDetection

PaddlePaddle / PaddleDetection
大约 1 年 前同步成功

通知 695
Star 11112
Fork 2696
P
PaddleDetection
未验证 提交 6063d163 编写于 作者: W wangguanzhong 提交者： GitHub
浏览文件
操作

unify int32 in op type (#3467)

上级 a8015e4b
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
ppdet/modeling/ops.py
浏览文件 @ 6063d163
...@@ -993,7 +993,7 @@ def multiclass_nms(bboxes, ...@@ -993,7 +993,7 @@ def multiclass_nms(bboxes,
else: else:
output = helper.create_variable_for_type_inference(dtype=bboxes.dtype) output = helper.create_variable_for_type_inference(dtype=bboxes.dtype)
index = helper.create_variable_for_type_inference(dtype='int') index = helper.create_variable_for_type_inference(dtype='int32')
inputs = {'BBoxes': bboxes, 'Scores': scores} inputs = {'BBoxes': bboxes, 'Scores': scores}
outputs = {'Out': output, 'Index': index} outputs = {'Out': output, 'Index': index}
...@@ -1136,7 +1136,7 @@ def matrix_nms(bboxes, ...@@ -1136,7 +1136,7 @@ def matrix_nms(bboxes,
else: else:
helper = LayerHelper('matrix_nms', **locals()) helper = LayerHelper('matrix_nms', **locals())
output = helper.create_variable_for_type_inference(dtype=bboxes.dtype) output = helper.create_variable_for_type_inference(dtype=bboxes.dtype)
index = helper.create_variable_for_type_inference(dtype='int') index = helper.create_variable_for_type_inference(dtype='int32')
outputs = {'Out': output, 'Index': index} outputs = {'Out': output, 'Index': index}
if return_rois_num: if return_rois_num:
rois_num = helper.create_variable_for_type_inference(dtype='int32') rois_num = helper.create_variable_for_type_inference(dtype='int32')
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或