ONNX 模型库
返回模型

说明文档

这些模型在 deepghs/anime_face_detection 和开源的真实照片数据集上训练。

因此同时支持动漫和照片。

配合 dghs-realutils 使用

pip install dghs-realutils
from realutils.detect import detect_faces

print(detect_faces('yolo/solo.jpg'))
# [((157, 94, 252, 208), 'face', 0.8836570382118225)]
print(detect_faces('yolo/2girls.jpg'))
# [((718, 154, 1110, 728), 'face', 0.8841166496276855), ((157, 275, 519, 715), 'face', 0.8668240904808044)]
print(detect_faces('yolo/3+cosplay.jpg'))
# [((349, 227, 413, 305), 'face', 0.8543888330459595), ((383, 61, 432, 117), 'face', 0.8080574870109558), ((194, 107, 245, 162), 'face', 0.8035706877708435)]
print(detect_faces('yolo/multiple.jpg'))
# [((1070, 728, 1259, 985), 'face', 0.8765808939933777), ((548, 286, 760, 558), 'face', 0.8693087697029114), ((896, 315, 1067, 520), 'face', 0.8671919107437134), ((1198, 220, 1342, 406), 'face', 0.8485829830169678), ((1376, 526, 1546, 719), 'face', 0.8469308018684387)]

更多信息,请参阅 realutils 文档

模型 类型 FLOPS 参数量 F1 分数 阈值 精确率(B) 召回率(B) mAP50(B) mAP50-95(B) F1 曲线 混淆矩阵 标签
face_detect_v0_s_yv12 yolo 21.5G 9.25M 0.74 0.272 0.86931 0.6404 0.73074 0.42652 plot confusion face
face_detect_v0_n_yv12 yolo 6.48G 2.57M 0.7 0.258 0.85246 0.59089 0.6793 0.39182 plot confusion face
face_detect_v0_l_yv11 yolo 87.3G 25.3M 0.77 0.291 0.88458 0.67474 0.76666 0.45722 plot confusion face
face_detect_v0_m_yv11 yolo 68.2G 20.1M 0.76 0.262 0.87947 0.67315 0.76073 0.45288 plot confusion face
face_detect_v0_s_yv11 yolo 21.5G 9.43M 0.73 0.271 0.87001 0.63572 0.72683 0.42706 plot confusion face
face_detect_v0_n_yv11 yolo 6.44G 2.59M 0.7 0.263 0.86044 0.58577 0.67641 0.38975 plot confusion face
face_detect_v0_l yolo 165G 43.6M 0.76 0.277 0.87894 0.67335 0.76313 0.4532 plot confusion face
face_detect_v0_m yolo 79.1G 25.9M 0.75 0.277 0.87687 0.66265 0.75114 0.44262 plot confusion face
face_detect_v0_s yolo 28.6G 11.1M 0.73 0.282 0.86932 0.63557 0.72494 0.42219 plot confusion face
face_detect_v0_n yolo 8.19G 3.01M 0.7 0.257 0.85337 0.58877 0.67471 0.38692 plot confusion face

deepghs/real_face_detection

作者 deepghs

object-detection dghs-imgutils
↓ 0 ♥ 2

创建时间: 2025-02-06 02:14:02+00:00

更新时间: 2025-02-21 09:14:14+00:00

在 Hugging Face 上查看

文件 (162)

.gitattributes
README.md
face_detect_v0_l/F1_curve.png
face_detect_v0_l/PR_curve.png
face_detect_v0_l/P_curve.png
face_detect_v0_l/R_curve.png
face_detect_v0_l/confusion_matrix.png
face_detect_v0_l/confusion_matrix_normalized.png
face_detect_v0_l/events.out.tfevents.1738779723.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.3155866.0
face_detect_v0_l/labels.jpg
face_detect_v0_l/labels.json
face_detect_v0_l/labels_correlogram.jpg
face_detect_v0_l/model.onnx ONNX
face_detect_v0_l/model.pt
face_detect_v0_l/model_type.json
face_detect_v0_l/results.csv
face_detect_v0_l/results.png
face_detect_v0_l/threshold.json
face_detect_v0_l_yv11/F1_curve.png
face_detect_v0_l_yv11/PR_curve.png
face_detect_v0_l_yv11/P_curve.png
face_detect_v0_l_yv11/R_curve.png
face_detect_v0_l_yv11/confusion_matrix.png
face_detect_v0_l_yv11/confusion_matrix_normalized.png
face_detect_v0_l_yv11/events.out.tfevents.1738787854.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.2342397.0
face_detect_v0_l_yv11/labels.jpg
face_detect_v0_l_yv11/labels.json
face_detect_v0_l_yv11/labels_correlogram.jpg
face_detect_v0_l_yv11/model.onnx ONNX
face_detect_v0_l_yv11/model.pt
face_detect_v0_l_yv11/model_type.json
face_detect_v0_l_yv11/results.csv
face_detect_v0_l_yv11/results.png
face_detect_v0_l_yv11/threshold.json
face_detect_v0_m/F1_curve.png
face_detect_v0_m/PR_curve.png
face_detect_v0_m/P_curve.png
face_detect_v0_m/R_curve.png
face_detect_v0_m/confusion_matrix.png
face_detect_v0_m/confusion_matrix_normalized.png
face_detect_v0_m/events.out.tfevents.1738772374.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.3036509.0
face_detect_v0_m/labels.jpg
face_detect_v0_m/labels.json
face_detect_v0_m/labels_correlogram.jpg
face_detect_v0_m/model.onnx ONNX
face_detect_v0_m/model.pt
face_detect_v0_m/model_type.json
face_detect_v0_m/results.csv
face_detect_v0_m/results.png
face_detect_v0_m/threshold.json
face_detect_v0_m_yv11/F1_curve.png
face_detect_v0_m_yv11/PR_curve.png
face_detect_v0_m_yv11/P_curve.png
face_detect_v0_m_yv11/R_curve.png
face_detect_v0_m_yv11/confusion_matrix.png
face_detect_v0_m_yv11/confusion_matrix_normalized.png
face_detect_v0_m_yv11/events.out.tfevents.1738776765.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.2155639.0
face_detect_v0_m_yv11/labels.jpg
face_detect_v0_m_yv11/labels.json
face_detect_v0_m_yv11/labels_correlogram.jpg
face_detect_v0_m_yv11/model.onnx ONNX
face_detect_v0_m_yv11/model.pt
face_detect_v0_m_yv11/model_type.json
face_detect_v0_m_yv11/results.csv
face_detect_v0_m_yv11/results.png
face_detect_v0_m_yv11/threshold.json
face_detect_v0_n/F1_curve.png
face_detect_v0_n/PR_curve.png
face_detect_v0_n/P_curve.png
face_detect_v0_n/R_curve.png
face_detect_v0_n/confusion_matrix.png
face_detect_v0_n/confusion_matrix_normalized.png
face_detect_v0_n/events.out.tfevents.1738769354.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.2987077.0
face_detect_v0_n/labels.jpg
face_detect_v0_n/labels.json
face_detect_v0_n/labels_correlogram.jpg
face_detect_v0_n/model.onnx ONNX
face_detect_v0_n/model.pt
face_detect_v0_n/model_type.json
face_detect_v0_n/results.csv
face_detect_v0_n/results.png
face_detect_v0_n/threshold.json
face_detect_v0_n_yv11/F1_curve.png
face_detect_v0_n_yv11/PR_curve.png
face_detect_v0_n_yv11/P_curve.png
face_detect_v0_n_yv11/R_curve.png
face_detect_v0_n_yv11/confusion_matrix.png
face_detect_v0_n_yv11/confusion_matrix_normalized.png
face_detect_v0_n_yv11/events.out.tfevents.1738769495.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.2029186.0
face_detect_v0_n_yv11/labels.jpg
face_detect_v0_n_yv11/labels.json
face_detect_v0_n_yv11/labels_correlogram.jpg
face_detect_v0_n_yv11/model.onnx ONNX
face_detect_v0_n_yv11/model.pt
face_detect_v0_n_yv11/model_type.json
face_detect_v0_n_yv11/results.csv
face_detect_v0_n_yv11/results.png
face_detect_v0_n_yv11/threshold.json
face_detect_v0_n_yv12/F1_curve.png
face_detect_v0_n_yv12/PR_curve.png
face_detect_v0_n_yv12/P_curve.png
face_detect_v0_n_yv12/R_curve.png
face_detect_v0_n_yv12/confusion_matrix.png
face_detect_v0_n_yv12/confusion_matrix_normalized.png
face_detect_v0_n_yv12/events.out.tfevents.1740109002.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.2806465.0
face_detect_v0_n_yv12/labels.jpg
face_detect_v0_n_yv12/labels.json
face_detect_v0_n_yv12/labels_correlogram.jpg
face_detect_v0_n_yv12/model.onnx ONNX
face_detect_v0_n_yv12/model.pt
face_detect_v0_n_yv12/model_type.json
face_detect_v0_n_yv12/results.csv
face_detect_v0_n_yv12/results.png
face_detect_v0_n_yv12/threshold.json
face_detect_v0_s/F1_curve.png
face_detect_v0_s/PR_curve.png
face_detect_v0_s/P_curve.png
face_detect_v0_s/R_curve.png
face_detect_v0_s/confusion_matrix.png
face_detect_v0_s/confusion_matrix_normalized.png
face_detect_v0_s/events.out.tfevents.1738722581.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.2211296.0
face_detect_v0_s/labels.jpg
face_detect_v0_s/labels.json
face_detect_v0_s/labels_correlogram.jpg
face_detect_v0_s/model.onnx ONNX
face_detect_v0_s/model.pt
face_detect_v0_s/model_type.json
face_detect_v0_s/results.csv
face_detect_v0_s/results.png
face_detect_v0_s/threshold.json
face_detect_v0_s_yv11/F1_curve.png
face_detect_v0_s_yv11/PR_curve.png
face_detect_v0_s_yv11/P_curve.png
face_detect_v0_s_yv11/R_curve.png
face_detect_v0_s_yv11/confusion_matrix.png
face_detect_v0_s_yv11/confusion_matrix_normalized.png
face_detect_v0_s_yv11/events.out.tfevents.1738772781.74aa4261331bbcab0987b94433af8ac622a654e7594a46e49f97167a.2086780.0
face_detect_v0_s_yv11/labels.jpg
face_detect_v0_s_yv11/labels.json
face_detect_v0_s_yv11/labels_correlogram.jpg
face_detect_v0_s_yv11/model.onnx ONNX
face_detect_v0_s_yv11/model.pt
face_detect_v0_s_yv11/model_type.json
face_detect_v0_s_yv11/results.csv
face_detect_v0_s_yv11/results.png
face_detect_v0_s_yv11/threshold.json
face_detect_v0_s_yv12/F1_curve.png
face_detect_v0_s_yv12/PR_curve.png
face_detect_v0_s_yv12/P_curve.png
face_detect_v0_s_yv12/R_curve.png
face_detect_v0_s_yv12/confusion_matrix.png
face_detect_v0_s_yv12/confusion_matrix_normalized.png
face_detect_v0_s_yv12/events.out.tfevents.1740122353.dbd7b05c6be69b3cad7adc4be45b0f3ae9bfa42b4e0c6f4f253fe7cb.3025198.0
face_detect_v0_s_yv12/labels.jpg
face_detect_v0_s_yv12/labels.json
face_detect_v0_s_yv12/labels_correlogram.jpg
face_detect_v0_s_yv12/model.onnx ONNX
face_detect_v0_s_yv12/model.pt
face_detect_v0_s_yv12/model_type.json
face_detect_v0_s_yv12/results.csv
face_detect_v0_s_yv12/results.png
face_detect_v0_s_yv12/threshold.json