marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
MIT License
1.38k stars 343 forks source link

customModels.md #538

Open 2509afroz opened 1 month ago

2509afroz commented 1 month ago

@marcoslucianops Hello, I've employed YOLOv5 for training on a GPU using custom data, then exported it to obtain a .onnx file. I aimed to implement this workflow on a Jetson Nano Orin for inference. However, during inference, I encountered an issue where only one label is displayed from the labels.txt file, while other objects are not detected. Moreover, the bounding boxes seem to encapsulate the entire image, resembling a single object detection, such as a person bounding box.