search

PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.
https://qiita.com/PINTO
MIT License
3.62k stars 574 forks source link

yolov11 face detection model #436

Open sasakiyudai opened 1 month ago

sasakiyudai commented 1 month ago

Issue Type

Others

OS

Mac OS

OS architecture

x86_64

Programming Language

Python

Framework

PyTorch, ONNX

Model name and Weights/Checkpoints URL

https://github.com/akanametov/yolo-face

Description

I would be happy if you could make a face detection model using the recently released yolov11 :)

Relevant Log Output

No response

URL or source code for simple inference testing code

No response

henryruhs commented 1 week ago

I tried to export it but the face landmarks are missing on the export. Seems to be an issue with ultralytics which ignores type=pose and keeps on using type=detect.

On pypi the older ultralytics releases have been wiped, so no quick way trying these.

The ONNX on the repository lack of the face landmarks as well, I was in contact with the maintainer:

Unfortunately, I have no ideas how to fix it, usually I work with detection only models.