未验证 提交 461f4cb4 编写于 作者: J Jason 提交者: GitHub

Merge pull request #141 from jiangjiajun/develop

fix bug for mtcnn
......@@ -591,7 +591,7 @@ class TFOpMapper(OpMapper):
# to change [192, -1]->[-1, 192], allways put -1 in the first dimension
# optimization for Paddle-Lite
in_shape = input.out_shapes[0]
if is_variable and in_shape.count(-1) < 1:
if not is_variable and in_shape.count(-1) < 1:
total_size = 1
for i in range(len(in_shape)):
total_size *= in_shape[i]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册