search

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.44k stars 355 forks source link

Support for pp-yolo model #83

Closed Justin-king-de closed 2 years ago

Justin-king-de commented 3 years ago

Hello @marcoslucianops @satchelwu @pieris98 I have read that pp-yolo gives good accuracy with very good FPS. But how to run pp-yolo model using deepstream SDK. How to convert pp-yolo model from paddle-paddle framework into deepstream compatible tensorrt model. Can you please help me in implementing this?

marcoslucianops commented 3 years ago

I'm planning to make YOLOR compatible with DeepStream in next months. It seems better than YOLOv4, YOLOv5 and PP-YOLO. About the implementation, you need to make a convertor for it and the output parsers.

Justin-king-de commented 3 years ago

@marcoslucianops Okay. But is there any tiny version of YOLOR? Because I have to run it on nano.

marcoslucianops commented 3 years ago

@marcoslucianops Okay. But is there any tiny version of YOLOR? Because I have to run it on nano.

It doesn't have for now. I will see about PP-YOLO.

Justin-king-de commented 3 years ago

@marcoslucianops Okay. Thanks. Please let me know when you are able to do it.

marcoslucianops commented 3 years ago

I will use these new models in my new projects, so I'm planning to start in next days.

Justin-king-de commented 3 years ago

Okay. Sure. I will post a query here about your progress after few weeks or a month.

garaujo23 commented 2 years ago

I will use these new models in my new projects, so I'm planning to start in next days.

Hey @marcoslucianops is there any update on YOLOR support for your deepstream repo?

marcoslucianops commented 2 years ago

End of year probably @garaujo23.

marcoslucianops commented 2 years ago

@garaujo23 Repo updated with native support for YOLOR models (for now, available for YOLOR-CSP, YOLOR-CSP,YOLOR-CSP-X and YOLOR-CSP-X).

garaujo23 commented 2 years ago

@marcoslucianops you are a legend, will do some testing now :)

marcoslucianops commented 2 years ago

Added support for PP-YOLOE.