ComfyUI-Dwpose-Tensorrt

ComfyUI Dwpose TensorRT ⚡

python cuda trt by-nc-sa/4.0

This project provides a Tensorrt implementation of Dwpose for ultra fast pose estimation inside ComfyUI

⭐ Support

If you like my projects and wish to see updates and new features, please consider supporting me. It helps a lot!

ComfyUI-Depth-Anything-Tensorrt ComfyUI-Upscaler-Tensorrt ComfyUI-Dwpose-Tensorrt ComfyUI-Rife-Tensorrt

ComfyUI-Whisper ComfyUI_InvSR ComfyUI-Thera ComfyUI-Video-Depth-Anything ComfyUI-PiperTTS

buy-me-coffees paypal-donation

⏱️ Performance

Note: The following results were benchmarked inside ComfyUI, using 1000 similar frames

Device	FPS
L40s	20
RTX4090	18.7

🚀 Installation

Navigate to the ComfyUI /custom_nodes directory

git clone https://github.com/yuvraj108c/ComfyUI-Dwpose-Tensorrt
cd ./ComfyUI-Dwpose-Tensorrt
pip install -r requirements.txt

☀️ Usage

• Insert node by Right Click -> tensorrt -> Dwpose Tensorrt
• Tensorrt engines will be built automatically
• Set DWPOSE precision to FP32 for better accuracy

🚨 Updates

3 May 2025

• Automatic tensorrt engines are built from the workflow itself, to simplify the process for non-technical people

🤖 Environment tested

• Ubuntu 22.04 LTS, Cuda 12.4, Tensorrt 10.2.0.post1, Python 3.10, L40s GPU
• Windows (Not tested, but should work)

👏 Credits

• https://github.com/IDEA-Research/DWPose
• https://github.com/legraphista/dwpose-video

License

Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0)