提交 cf2ed179 编写于 作者: C chengduoZH

fix prior_op unit test

上级 282dadb0
...@@ -65,9 +65,9 @@ class TestPriorBoxOp(OpTest): ...@@ -65,9 +65,9 @@ class TestPriorBoxOp(OpTest):
self.batch_size = 10 self.batch_size = 10
self.min_sizes = [2, 4] self.min_sizes = [2, 4]
self.min_sizes = np.array(self.min_sizes).astype('float32') self.min_sizes = np.array(self.min_sizes).astype('float32').tolist()
self.max_sizes = [5, 10] self.max_sizes = [5, 10]
self.max_sizes = np.array(self.max_sizes).astype('float32') self.max_sizes = np.array(self.max_sizes).astype('float32').tolist()
self.aspect_ratios = [2.0, 3.0] self.aspect_ratios = [2.0, 3.0]
self.flip = True self.flip = True
self.real_aspect_ratios = [1, 2.0, 1.0 / 2.0, 3.0, 1.0 / 3.0] self.real_aspect_ratios = [1, 2.0, 1.0 / 2.0, 3.0, 1.0 / 3.0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册