detect_video have problem

[hh-kai]

version 0.0.6

use like this for video
python trt.py -e yolov8n.trt -v src/video1.mp4 --end2end

run failed
Traceback (most recent call last): File "trt.py", line 35, in <module> pred.detect_video(video, conf=0.1, end2end=args.end2end) # set 0 use a webcam File "/home/jetson-nx/share/TensorRT-For-YOLO-Series-0.0.6/utils/utils.py", line 118, in detect_video dets = np.concatenate([np.array(final_boxes)[:int(num[0])], np.array(final_scores)[:int(num[0])], np.array(final_cls_inds)[:int(num[0])]], axis=-1) File "<__array_function__ internals>", line 200, in concatenate ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 0, the array at index 0 has size 16 and the array at index 1 has size 1

i use AI fix

then ok

[Linaom1214]

version 0.0.6

use like this for video python trt.py -e yolov8n.trt -v src/video1.mp4 --end2end

run failed Traceback (most recent call last): File "trt.py", line 35, in <module> pred.detect_video(video, conf=0.1, end2end=args.end2end) # set 0 use a webcam File "/home/jetson-nx/share/TensorRT-For-YOLO-Series-0.0.6/utils/utils.py", line 118, in detect_video dets = np.concatenate([np.array(final_boxes)[:int(num[0])], np.array(final_scores)[:int(num[0])], np.array(final_cls_inds)[:int(num[0])]], axis=-1) File "<__array_function__ internals>", line 200, in concatenate ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 0, the array at index 0 has size 16 and the array at index 1 has size 1

i use AI fix

then ok

you can create a pr, thanks