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.47k stars 357 forks source link

Does this code support yolo-pose？ #309

Open lofyol opened 1 year ago

marcoslucianops commented 1 year ago

It doesn't support pose yet.

hhackbarth commented 1 year ago

I am also interested in adaptions for YOLOv8-pose. Anyone who has gone that path already and can share something?

simonvvd commented 1 year ago

@marcoslucianops Could you give the steps to add support for yolov8-pose to this project? We're willing to make a PR for it as we need it for our product.

marcoslucianops commented 1 year ago

The pose estimation needs a custom output layer using the TensorMeta. It's more complex.

marcoslucianops commented 1 year ago

Added support for Pose: https://github.com/marcoslucianops/DeepStream-Yolo-Pose