search

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.26k stars 196 forks source link

【utils/yolo.cpp】代码请教 #65

Closed MMmingming closed 6 months ago

MMmingming commented 6 months ago

你好，请问utils/yolo.cpp中，这里的m_output_objects_width = 7 如何理解。 如果是每个目标信息应该等于 4+ cls的大小吗，虽然跑的结果正确但这里不太理解，希望能够解答谢谢！

https://github.com/FeiYull/TensorRT-Alpha/blob/ade5898a58e10580d555a3ded5fa630ec2eeee70/utils/yolo.cpp#L21C23-L21C23

FeiYull commented 6 months ago

@MMmingming 7:left, top, right, bottom, confidence, class, keepflag;

MMmingming commented 6 months ago

@FeiYull thanks！