提交 00e822d2 编写于 作者: D dengkaipeng

fix test_detection. test=develop

上级 abb5a9c7
......@@ -484,7 +484,7 @@ class TestYoloDetection(unittest.TestCase):
program = Program()
with program_guard(program):
x = layers.data(name='x', shape=[30, 7, 7], dtype='float32')
img_size = layers.data(name='x', shape=[2], dtype='int32')
img_size = layers.data(name='img_size', shape=[2], dtype='int32')
boxes, scores = layers.yolo_box(x, img_size, [10, 13, 30, 13], 10,
0.01, 32)
self.assertIsNotNone(boxes)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册