基于 WildChlamydia/MiVOLO 封装的人脸/人体年龄与性别识别模型,作为 funmodel 命名空间下的 funmodel.mivolo 子包发布,提供开箱即用的预测器(自动下载权重、YOLO 检测 + MiVOLO 分类)。
uv add funmodel-mivolo
# 或
pip install funmodel-mivoloimport 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 是一个专注于实用工具库的开源组织, 涵盖云存储、数据处理、AI、多媒体与开发工具链等方向。
- 🏠 组织主页:https://github.com/farfarfun
- 📦 PyPI:https://pypi.org/user/niuliangtao/
- 📧 联系:farfarfun@qq.com
本项目基于 MIT 协议开源。