提交 2ead2931 编写于 作者: M MRXLT

update infer lib && fix blaze postprocess

上级 57612da6
...@@ -31,7 +31,7 @@ message( "WITH_GPU = ${WITH_GPU}") ...@@ -31,7 +31,7 @@ message( "WITH_GPU = ${WITH_GPU}")
# Paddle Version should be one of: # Paddle Version should be one of:
# latest: latest develop build # latest: latest develop build
# version number like 1.5.2 # version number like 1.5.2
SET(PADDLE_VERSION "1.7.2") SET(PADDLE_VERSION "1.8.4")
if (WITH_GPU) if (WITH_GPU)
SET(PADDLE_LIB_VERSION "${PADDLE_VERSION}-gpu-cuda${CUDA_VERSION_MAJOR}-cudnn7-avx-mkl") SET(PADDLE_LIB_VERSION "${PADDLE_VERSION}-gpu-cuda${CUDA_VERSION_MAJOR}-cudnn7-avx-mkl")
......
...@@ -420,7 +420,7 @@ class RCNNPostprocess(object): ...@@ -420,7 +420,7 @@ class RCNNPostprocess(object):
for key in image_with_bbox: for key in image_with_bbox:
if key == "image": if key == "image":
continue continue
if ".lod" in key: if ".lod" in key or "im_shape" in key:
continue continue
fetch_name = key fetch_name = key
bbox_result = self._get_bbox_result(image_with_bbox, fetch_name, bbox_result = self._get_bbox_result(image_with_bbox, fetch_name,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册