Paddle hub solov2 [operator < gather > error]

JunhuaY commented 2 years ago

欢迎您反馈PaddleHub使用问题，非常感谢您对PaddleHub的贡献！在留下您的问题时，辛苦您同步提供如下信息：

版本、环境信息 1）PaddleHub和PaddlePaddle版本：请提供您的PaddleHub和PaddlePaddle版本号，例如PaddleHub1.4.1，PaddlePaddle1.6.2 2）系统环境：请您描述系统类型，例如Linux/Windows/MacOS/，python版本
复现信息：如为报错，请给出复现环境、复现步骤

PaddleHub 2.1.1 与 PaddlePaddle 2.2.0 安装在windows10 anaconda上时运行solov2 示例程序

import cv2 import paddlehub as hub

img = cv2.imread('/PATH/TO/IMAGE') model = hub.Module(name='solov2', use_gpu=False) output = model.predict(image=img, visualization=False)

出现报错 W1111 19:56:47.858419 4732 analysis_predictor.cc:1353] Deprecated. Please use CreatePredictor instead. e[37m--- Fused 0 subgraphs into layer_norm op.e[0m e[37m--- fused 0 pairs of fc gru patternse[0m Traceback (most recent call last): File "tes.py", line 7, in output = model.predict(image=img, visualization=True) File "C:\Users\mushroom\anaconda3\envs\paddle_env\lib\site-packages\paddlehub\compat\paddle_utils.py", line 220, in runner return func(*args, **kwargs) File "C:\Users\mushroom.paddlehub\modules\solov2\module.py", line 150, in predict self.predictor.zero_copy_run() IndexError: In user code:

File "tools/export_model.py", line 89, in <module>
  main()
File "tools/export_model.py", line 63, in main
  test_fetches = model.test(feed_vars, exclude_nms=FLAGS.exclude_nms)
File "/paddle/tools/PaddleDetection/ppdet/modeling/architectures/solov2.py", line 175, in test
  return self.build(feed_vars, mode='test')
File "/paddle/tools/PaddleDetection/ppdet/modeling/architectures/solov2.py", line 110, in build
  return self.bbox_head.get_prediction(*seg_inputs)
File "/paddle/tools/PaddleDetection/ppdet/modeling/anchor_heads/solov2_head.py", line 336, in get_prediction
  im_info[idx])
File "/paddle/tools/PaddleDetection/ppdet/modeling/anchor_heads/solov2_head.py", line 382, in get_seg_single
  strides = fluid.layers.gather(strides, index=inds[:, 0])
File "/usr/local/lib/python3.7/site-packages/paddle/fluid/layers/nn.py", line 8328, in gather
  attrs={'overwrite': overwrite})
File "/usr/local/lib/python3.7/site-packages/paddle/fluid/layer_helper.py", line 43, in append_op
  return self.main_program.current_block().append_op(*args, **kwargs)
File "/usr/local/lib/python3.7/site-packages/paddle/fluid/framework.py", line 2917, in append_op
  attrs=kwargs.get("attrs", None))
File "/usr/local/lib/python3.7/site-packages/paddle/fluid/framework.py", line 2014, in __init__
  for frame in traceback.extract_stack():

OutOfRangeError: The element of Index must be less than the size of input dim size of axis which is 3872, but received index element which is 3872 in the 20 index.
  [Hint: Expected p_index[i] < input_size, but received p_index[i]:3872 >= input_size:3872.] (at C:\home\workspace\Paddle_release\paddle/fluid/operators/gather.h:82)
  [operator < gather > error]

尝试更换目标图片仍然同样报错,应该与图片本身无关