Skip to content

About

基于 DWPose ONNX 模型的全身姿态估计推理包:提供 DWposePredict,支持本地图片、图片 URL 与摄像头视频流的人体关键点检测及骨架绘制。

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Repository files navigation

funmodel-dwpose

基于 DWPose(含 CMU OpenPose / pytorch-openpose 移植代码)封装的人体全身姿态估计模型,作为 funmodel 命名空间下的一个子包发布(import funmodel.dwpose),提供开箱即用的关键点检测与骨架绘制能力。

安装

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

最小示例

import cv2

from funmodel.dwpose import DWposePredict

predictor = DWposePredict()
image = cv2.imread("input.jpg")
result, drawn_image = predictor.predict(image, draw=True)
cv2.imwrite("output.jpg", drawn_image)

首次调用 DWposePredict() 会自动从对象存储下载 ONNX 权重(yolox_l.onnx、dw-ll_ucoco_384.onnx)到本地缓存目录(~/.funmodel/<model_name>/)。

许可证

本仓库不是自研实现,funmodel/dwpose/core/ 下的 5 个模块全部取自 IDEA-Research/DWPose(Apache-2.0)。 因此本仓库沿用上游协议,以 Apache License 2.0 开源,而不是组织默认的 MIT; 上游署名见 NOTICE。

第三方代码来源

funmodel/dwpose/core/ 下的检测、姿态估计与绘制代码取自:

前四项为 Apache-2.0,其署名义务已通过 LICENSE、NOTICE 与逐文件版权头履行。 后两项的许可证状况无法通过署名满足:pytorch-openpose 未声明许可证,CMU OpenPose 的协议禁止分发且不授予再授权权。该部分的处置(含 PyPI 上已发布版本)在 farfarfun/todo-list#894 跟踪, 尚未结论。逐项实测证据与许可证原文引用见 THIRD_PARTY_LICENSES.md,引用方请据此自行评估。


关于 farfarfun

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

本项目基于 Apache-2.0 协议开源(衍生自第三方 Apache-2.0 代码,沿用上游协议, 非组织默认的 MIT;详见上方「许可证」与 NOTICE)。

About

基于 DWPose ONNX 模型的全身姿态估计推理包:提供 DWposePredict,支持本地图片、图片 URL 与摄像头视频流的人体关键点检测及骨架绘制。

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Used by

Contributors

Languages