laugh12321 / TensorRT-YOLO

🚀 你的YOLO部署神器。TensorRT Plugin、CUDA Kernel、CUDA Graphs三管齐下,享受闪电般的推理速度。| Your YOLO Deployment Powerhouse. With the synergy of TensorRT Plugins, CUDA Kernels, and CUDA Graphs, experience lightning-fast inference speeds.
https://github.com/laugh12321/TensorRT-YOLO
GNU General Public License v3.0
622 stars 73 forks source link

[Help]: IndexError: list index out of range #49

Closed wk565 closed 2 weeks ago

wk565 commented 2 weeks ago

xxxxx/python3.9/site-packages/tensorrt_yolo/infer/visualize.py", line 122, in visualizedetections color = labels[class][1] IndexError: list index out of range,我的模型只检测一个类别

wk565 commented 2 weeks ago

解决了,我没有在txt中设置类别