diff --git a/README.md b/README.md index 4741a4e39939e4f0ec9e3a4035fe68b7d8cf6dce..cc67fb5e0d09b205232df198f2ae21575c21f8d3 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,7 @@ X2Paddle在之前的开发过程中,对比了PaddlePaddle与主流框架上的 ## 环境依赖 python == 2.7 | python >= 3.5 -paddlepaddle >= 1.5.0 +paddlepaddle >= 1.6.1 **按需安装以下依赖** tensorflow : tensorflow == 1.14.0 diff --git a/x2paddle/decoder/tf_decoder.py b/x2paddle/decoder/tf_decoder.py index 0bbfc3e006d53ba6001fd37fd16fe7a729f5ad91..9bf45b408393cc5609387e06eb3bb87b2dfd51aa 100644 --- a/x2paddle/decoder/tf_decoder.py +++ b/x2paddle/decoder/tf_decoder.py @@ -390,7 +390,7 @@ class TFDecoder(object): compare12 = (results[1] == results[2]) compare = compare01 & compare12 - index = numpy.argwhere(compare==False).flatten() + index = numpy.argwhere(compare == False).flatten() results[0][index] = -1 return results[0].tolist() @@ -421,6 +421,6 @@ class TFDecoder(object): return shape[0].tolist() compare = compare01 & compare12 - index = numpy.argwhere(compare==False).flatten() + index = numpy.argwhere(compare == False).flatten() shapes[0][index] = -1 return shapes[0].tolist()