提交 deec3ac1 编写于 作者: D dengkaipeng

fix infer error.

上级 c9d4676b
......@@ -655,8 +655,6 @@ def yolo_box(x,
raise TypeError("Input img_size of yolo_box must be Variable")
if not isinstance(anchors, list) and not isinstance(anchors, tuple):
raise TypeError("Attr anchors of yolo_box must be list or tuple")
if not isinstance(anchor_mask, list) and not isinstance(anchor_mask, tuple):
raise TypeError("Attr anchor_mask of yolo_box must be list or tuple")
if not isinstance(class_num, int):
raise TypeError("Attr class_num of yolo_box must be an integer")
if not isinstance(conf_thresh, float):
......@@ -668,7 +666,7 @@ def yolo_box(x,
attrs = {
"anchors": anchors,
"class_num": class_num,
"conf_thresh": ignore_thresh,
"conf_thresh": conf_thresh,
"downsample_ratio": downsample_ratio,
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册