使DiouNMS出现数组维度错误
收藏
在PaddleDetection的YOLOv3中使用MultiClassDiouNMS出现以下错误:
ValueError: all the input arrays must have same number of dimensions
At:
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/numpy/core/shape_base.py(340): hstack
/home/aistudio/work/PaddleDetection/ppdet/modeling/ops.py(737): _diou_nms
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/layers/nn.py(14192): __call__
F0818 12:10:41.363613 8453 exception_holder.h:37] std::exception caught, ValueError: all the input arrays must have same number of dimensions
ops.py(737) dets_j = np.hstack((scores_j[:, np.newaxis], rois_j)).astype(
np.float32, copy=False)
请问该如何解决?
0
收藏
请登录后评论