search

marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.1 / 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.5k stars 362 forks source link

YOLOV8-OBB #555

Open TianLuluC opened 4 months ago

TianLuluC commented 4 months ago

Hi, I recently tried to use deepstream to load the YOLOV8-OBB model, but I tried many times without success. I would like to ask, apart from the post-processing part (nvdsparsebbox_Yolo.cpp), are there any other places that need to be modified? I would be particularly grateful if you could give me some pointers.Thank you!Looking forward to your reply！@marcoslucianops

marcoslucianops commented 6 days ago

The structure of the DeepStream doesn't support OBB natively. You need to use tensor-meta and create the post-processing logic to use it.