deepghs/real_head_detection

By deepghs · ↓ 0 · ♥ 1 · Updated 2025-02-20

What's in this repo

This page lists the files in deepghs/real_head_detection, including model.onnx and more.

8 ONNX models (.onnx files)
131 files, including 56 .png, 25 .json, 16 .jpg, 8 .0, 8 .onnx, 8 .pt
Last updated 2025-02-20

ONNX files

File	Size	Download
head_detect_v0_l/model.onnx	—	Download
head_detect_v0_l_yv11/model.onnx	—	Download
head_detect_v0_m/model.onnx	—	Download
head_detect_v0_m_yv11/model.onnx	—	Download
head_detect_v0_n/model.onnx	—	Download
head_detect_v0_n_yv11/model.onnx	—	Download
head_detect_v0_s/model.onnx	—	Download
head_detect_v0_s_yv11/model.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

These models are trained on deepghs/anime_head_detection and open-sourced real photos datasets.

So both anime and photos are supported.

Use this with dghs-realutils

pip install dghs-realutils

from realutils.detect import detect_heads

print(detect_heads('yolo/solo.jpg'))
# [((162, 47, 305, 210), 'head', 0.7701659202575684)]
print(detect_heads('yolo/2girls.jpg'))
# [((683, 48, 1199, 754), 'head', 0.8410779237747192), ((105, 91, 570, 734), 'head', 0.8339194059371948)]
print(detect_heads('yolo/3+cosplay.jpg'))
# [((329, 194, 426, 309), 'head', 0.8123012781143188), ((359, 20, 448, 122), 'head', 0.8047150373458862), ((185, 81, 265, 166), 'head', 0.7797152996063232)]
print(detect_heads('yolo/multiple.jpg'))
# [((867, 259, 1084, 527), 'head', 0.8264595866203308), ((1364, 448, 1583, 724), 'head', 0.8254891633987427), ((480, 201, 781, 565), 'head', 0.8191508054733276), ((1189, 175, 1398, 412), 'head', 0.8097156286239624), ((1028, 671, 1277, 992), 'head', 0.8084591627120972)]

For more information, see documentation of realutils.

Model	Type	FLOPS	Params	F1 Score	Threshold	precision(B)	recall(B)	mAP50(B)	mAP50-95(B)	F1 Plot	Confusion	Labels
head_detect_v0_l_yv11	yolo	87.3G	25.3M	0.81	0.199	0.90226	0.72872	0.81049	0.5109	plot	confusion	`head`
head_detect_v0_m_yv11	yolo	68.2G	20.1M	0.8	0.206	0.89855	0.72654	0.80704	0.50804	plot	confusion	`head`
head_detect_v0_s_yv11	yolo	21.5G	9.43M	0.78	0.187	0.88726	0.69234	0.77518	0.47825	plot	confusion	`head`
head_detect_v0_n_yv11	yolo	6.44G	2.59M	0.74	0.14	0.87359	0.64011	0.73393	0.44118	plot	confusion	`head`
head_detect_v0_l	yolo	165G	43.6M	0.81	0.234	0.89921	0.74092	0.81715	0.51615	plot	confusion	`head`
head_detect_v0_m	yolo	79.1G	25.9M	0.8	0.228	0.90006	0.72646	0.80614	0.50586	plot	confusion	`head`
head_detect_v0_s	yolo	28.6G	11.1M	0.78	0.182	0.89224	0.69382	0.77804	0.48067	plot	confusion	`head`
head_detect_v0_n	yolo	8.19G	3.01M	0.74	0.172	0.8728	0.64823	0.73865	0.44501	plot	confusion	`head`

Model info

object-detection dghs-imgutils

dghs-imgutils tensorboard onnx art anime photo head object-detection dataset:deepghs/real_head_detection dataset:deepghs/anime_head_detection region:us

↓ 0 ♥ 1

Created: 2025-02-06

Updated: 2025-02-20

View on Hugging Face

Files (131)

.gitattributes

README.md

head_detect_v0_l/F1_curve.png

head_detect_v0_l/PR_curve.png

head_detect_v0_l/P_curve.png

head_detect_v0_l/R_curve.png

head_detect_v0_l/confusion_matrix.png

head_detect_v0_l/confusion_matrix_normalized.png

head_detect_v0_l/events.out.tfevents.1738807573.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.2673077.0

head_detect_v0_l/labels.jpg

head_detect_v0_l/labels.json

head_detect_v0_l/labels_correlogram.jpg

head_detect_v0_l/model.onnx ONNX

head_detect_v0_l/model.pt

head_detect_v0_l/model_type.json

head_detect_v0_l/results.csv

head_detect_v0_l/results.png

head_detect_v0_l/threshold.json

head_detect_v0_l_yv11/F1_curve.png

head_detect_v0_l_yv11/PR_curve.png

head_detect_v0_l_yv11/P_curve.png

head_detect_v0_l_yv11/R_curve.png

head_detect_v0_l_yv11/confusion_matrix.png

head_detect_v0_l_yv11/confusion_matrix_normalized.png

head_detect_v0_l_yv11/events.out.tfevents.1738833414.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.4022117.0

head_detect_v0_l_yv11/labels.jpg

head_detect_v0_l_yv11/labels.json

head_detect_v0_l_yv11/labels_correlogram.jpg

head_detect_v0_l_yv11/model.onnx ONNX

head_detect_v0_l_yv11/model.pt

head_detect_v0_l_yv11/model_type.json

head_detect_v0_l_yv11/results.csv

head_detect_v0_l_yv11/results.png

head_detect_v0_l_yv11/threshold.json

head_detect_v0_m/F1_curve.png

head_detect_v0_m/PR_curve.png

head_detect_v0_m/P_curve.png

head_detect_v0_m/R_curve.png

head_detect_v0_m/confusion_matrix.png

head_detect_v0_m/confusion_matrix_normalized.png

head_detect_v0_m/events.out.tfevents.1738697544.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.832744.0

head_detect_v0_m/labels.jpg

head_detect_v0_m/labels.json

head_detect_v0_m/labels_correlogram.jpg

head_detect_v0_m/model.onnx ONNX

head_detect_v0_m/model.pt

head_detect_v0_m/model_type.json

head_detect_v0_m/results.csv

head_detect_v0_m/results.png

head_detect_v0_m/threshold.json

head_detect_v0_m_yv11/F1_curve.png

head_detect_v0_m_yv11/PR_curve.png

head_detect_v0_m_yv11/P_curve.png

head_detect_v0_m_yv11/R_curve.png

head_detect_v0_m_yv11/confusion_matrix.png

head_detect_v0_m_yv11/confusion_matrix_normalized.png

head_detect_v0_m_yv11/events.out.tfevents.1738801529.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.3511507.0

head_detect_v0_m_yv11/labels.jpg

head_detect_v0_m_yv11/labels.json

head_detect_v0_m_yv11/labels_correlogram.jpg

head_detect_v0_m_yv11/model.onnx ONNX

head_detect_v0_m_yv11/model.pt

head_detect_v0_m_yv11/model_type.json

head_detect_v0_m_yv11/results.csv

head_detect_v0_m_yv11/results.png

head_detect_v0_m_yv11/threshold.json

head_detect_v0_n/F1_curve.png

head_detect_v0_n/PR_curve.png

head_detect_v0_n/P_curve.png

head_detect_v0_n/R_curve.png

head_detect_v0_n/confusion_matrix.png

head_detect_v0_n/confusion_matrix_normalized.png

head_detect_v0_n/events.out.tfevents.1738684051.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.610810.0

head_detect_v0_n/labels.jpg

head_detect_v0_n/labels.json

head_detect_v0_n/labels_correlogram.jpg

head_detect_v0_n/model.onnx ONNX

head_detect_v0_n/model.pt

head_detect_v0_n/model_type.json

head_detect_v0_n/results.csv

head_detect_v0_n/results.png

head_detect_v0_n/threshold.json

head_detect_v0_n_yv11/F1_curve.png

head_detect_v0_n_yv11/PR_curve.png

head_detect_v0_n_yv11/P_curve.png

head_detect_v0_n_yv11/R_curve.png

head_detect_v0_n_yv11/confusion_matrix.png

head_detect_v0_n_yv11/confusion_matrix_normalized.png

head_detect_v0_n_yv11/events.out.tfevents.1738685755.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.1608928.0

head_detect_v0_n_yv11/labels.jpg

head_detect_v0_n_yv11/labels.json

head_detect_v0_n_yv11/labels_correlogram.jpg

head_detect_v0_n_yv11/model.onnx ONNX

head_detect_v0_n_yv11/model.pt

head_detect_v0_n_yv11/model_type.json

head_detect_v0_n_yv11/results.csv

head_detect_v0_n_yv11/results.png

head_detect_v0_n_yv11/threshold.json

head_detect_v0_s/F1_curve.png

head_detect_v0_s/PR_curve.png

head_detect_v0_s/P_curve.png

head_detect_v0_s/R_curve.png

head_detect_v0_s/confusion_matrix.png

head_detect_v0_s/confusion_matrix_normalized.png

head_detect_v0_s/events.out.tfevents.1738662536.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.259603.0

head_detect_v0_s/labels.jpg

head_detect_v0_s/labels.json

head_detect_v0_s/labels_correlogram.jpg

head_detect_v0_s/model.onnx ONNX

head_detect_v0_s/model.pt

head_detect_v0_s/model_artifacts.json

head_detect_v0_s/model_type.json

head_detect_v0_s/results.csv

head_detect_v0_s/results.png

head_detect_v0_s/threshold.json

head_detect_v0_s_yv11/F1_curve.png

head_detect_v0_s_yv11/PR_curve.png

head_detect_v0_s_yv11/P_curve.png

head_detect_v0_s_yv11/R_curve.png

head_detect_v0_s_yv11/confusion_matrix.png

head_detect_v0_s_yv11/confusion_matrix_normalized.png

head_detect_v0_s_yv11/events.out.tfevents.1738660874.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.1207228.0

head_detect_v0_s_yv11/labels.jpg

head_detect_v0_s_yv11/labels.json

head_detect_v0_s_yv11/labels_correlogram.jpg

head_detect_v0_s_yv11/model.onnx ONNX

head_detect_v0_s_yv11/model.pt

head_detect_v0_s_yv11/model_type.json

head_detect_v0_s_yv11/results.csv

head_detect_v0_s_yv11/results.png

head_detect_v0_s_yv11/threshold.json

Related models

More object-detection models

Browse all object-detection models