├── .gitignore ├── LICENSE ├── __init__.py ├── assets └── demo.PNG ├── engine_builder_node.py ├── export_onnx.py ├── export_trt.py ├── readme.md ├── requirements.txt ├── trt_inference_video.py ├── utilities.py └── yolo_nas_pose ├── detection.py ├── pose_estimation.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | *.onnx -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/LICENSE -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/__init__.py -------------------------------------------------------------------------------- /assets/demo.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/assets/demo.PNG -------------------------------------------------------------------------------- /engine_builder_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/engine_builder_node.py -------------------------------------------------------------------------------- /export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/export_onnx.py -------------------------------------------------------------------------------- /export_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/export_trt.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tensorrt 2 | opencv-python 3 | polygraphy 4 | matplotlib -------------------------------------------------------------------------------- /trt_inference_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/trt_inference_video.py -------------------------------------------------------------------------------- /utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/utilities.py -------------------------------------------------------------------------------- /yolo_nas_pose/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/yolo_nas_pose/detection.py -------------------------------------------------------------------------------- /yolo_nas_pose/pose_estimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/yolo_nas_pose/pose_estimation.py -------------------------------------------------------------------------------- /yolo_nas_pose/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvraj108c/ComfyUI-YoloNasPose-Tensorrt/HEAD/yolo_nas_pose/utils.py --------------------------------------------------------------------------------