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 354 forks source link

Does this code support yolo-face? #326

Open knavezl opened 1 year ago

marcoslucianops commented 1 year ago

What's the repo?

phamminhhanhuet commented 1 year ago

No description provided.

Maybe this repo: https://github.com/deepcam-cn/yolov5-face ? It seems that the architecture of yolo-face (yolov5, yolov7) is same as the original yolov5, yolov7. I'm doing with yolo7-face too, model was converted to onnx & engine formats well (I modified the output of onnx model to (classes, boxes, scores, landmarks) but I'm finding where to extract the output detection info.

What's the repo?

marcoslucianops commented 1 year ago

This repo doesn't support landmarks, you should create a custom output parser to get the output from yolo-face. I can try to convert it in the future (I'm finishing some works and I will start the segmentation, pose, and other conversions for this repo).

marcoslucianops commented 1 year ago

https://github.com/marcoslucianops/DeepStream-Yolo-Face