Skip to content

Latest commit

 

History

26 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

funmodel-mivolo

基于 WildChlamydia/MiVOLO 封装的人脸/人体年龄与性别识别模型,作为 funmodel 命名空间下的 funmodel.mivolo 子包发布,提供开箱即用的预测器(自动下载权重、YOLO 检测 + MiVOLO 分类)。

安装

uv add funmodel-mivolo
# 或
pip install funmodel-mivolo

最小示例

import cv2
from funmodel.mivolo.predictor import MivoloPredictor

predictor = MivoloPredictor(device="cpu")
image = cv2.imread("example.jpg")
results, _ = predictor.predict(image)
print(results)  # [{"age": ..., "gender": ..., "gender_score": ..., ...}, ...]

首次调用会自动从组织 OSS 下载 mivolo_imbd.pth.tar 和 yolov8x_person_face.pt 两个权重文件到本地缓存目录。


关于 farfarfun

farfarfun 是一个专注于实用工具库的开源组织, 涵盖云存储、数据处理、AI、多媒体与开发工具链等方向。

本项目基于 MIT 协议开源。

About

基于 MiVOLO 的人脸年龄与性别识别推理包:封装 MivoloPredictor(YOLO 检测 + MiVOLO 分类),支持图片与摄像头流的年龄性别预测并可绘制结果。

Topics

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Used by

Contributors

Languages