hi MASTER , i try run this demo ( python test.py --data dataset/custon.yaml --model2_weight model/yolov4/weights/v4_best.pt --model1_weight model/yolov5/v5_best.pt --model2_cfg model/yolov4/cfg/yolov4-pacsp-x.cfg) , the details of the error are as follows.could you help solve it?Thanks
Weighted-Boxes-Fusion-implementation-main/test.py", line 307, in run
pred = np.concatenate([p_boxes, np.expand_dims(p_scores, axis=1), np.expand_dims(p_labels, axis=1)], axis=1)
File "", line 180, in concatenate
ValueError: all the input arrays must have same number of dimensions, but the array at index 0 has 2 dimension(s) and the array at index 2 has 3 dimension(s)
hi MASTER , i try run this demo ( python test.py --data dataset/custon.yaml --model2_weight model/yolov4/weights/v4_best.pt --model1_weight model/yolov5/v5_best.pt --model2_cfg model/yolov4/cfg/yolov4-pacsp-x.cfg) , the details of the error are as follows.could you help solve it?Thanks
Weighted-Boxes-Fusion-implementation-main/test.py", line 307, in run pred = np.concatenate([p_boxes, np.expand_dims(p_scores, axis=1), np.expand_dims(p_labels, axis=1)], axis=1) File "", line 180, in concatenate
ValueError: all the input arrays must have same number of dimensions, but the array at index 0 has 2 dimension(s) and the array at index 2 has 3 dimension(s)