ComfyUI Extension: ComfyUI Dwpose TensorRT

Authored by yuvraj108c

Created

Updated

29 stars

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

Custom Nodes (1)

README

<div align="center">

ComfyUI Dwpose TensorRT ⚡

python cuda trt by-nc-sa/4.0

</div>

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

<p align="center"> <img src="https://github.com/user-attachments/assets/65d8ad0e-2db8-43e3-8984-fbcedd272c12" height="400" /> </p>

⭐ 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)