Closed fatfishlin closed 2 years ago
The "-c" (or "--category_num") option is for specifying number of object classes for your custom model.
The error messages basically told you that you called "trt_yolo.py" with "-c 2" but the model outputs a class ID larger than or equal to 2 (while only 0 and 1 are valid class IDs).
hi,I do "Demo #5: YOLOv4" as you said,using: $python3 trt_yolo.py --usb 6 --width 1920 --height 1080 -m yolov4-416 it is ok(I have one camera in /dev/video6),detection is ok ,but I using: $python3 trt_yolo.py --usb 6 --width 1920 --height 1080 -m yolov4-416 -c 2 have error: [ WARN:0] global /home/nvidia/opencv/opencv-4.4.0/modules/videoio/src/cap_gstreamer.cpp (935) open OpenCV | GStreamer warning: Cannot query video position: status=0, value=-1, duration=-1 Traceback (most recent call last): File "trt_yolo.py", line 110, in
main()
File "trt_yolo.py", line 103, in main
loop_and_detect(cam, trt_yolo, conf_th=0.3, vis=vis)
File "trt_yolo.py", line 68, in loop_and_detect
img = vis.draw_bboxes(img, boxes, confs, clss)
File "/home/nvidia/dev_ws/src/tensorrt_demos-master/utils/visualization.py", line 96, in draw_bboxes
color = self.colors[cl]
IndexError: list index out of range
please help me. thanks.