feat: add CIGPose pose estimation models#260
Conversation
Signed-off-by: wep21 <[email protected]>
Inference Results (all variants, CUDA)Environment: NVIDIA GeForce RTX 4070 Laptop GPU (8 GB), driver 595.71.05, CUDA EP via ort v2.0.0-rc.12 Commandcargo run -r -F cuda-full --example pose-estimation -- cigpose \
--scale {l,x} --is-coco {true,false} --hires {true,false} \
--dtype {fp32,fp16} --device cuda:0 --processor-device cuda:0Performance
Times are per-batch averages over the 4 detected persons (3 dry-runs before timing). fp16 gives a ~1.5–1.7× inference speedup across all variants. All 10 runs (5 variants × fp32/fp16) completed successfully.
Annotated Outputs
🤖 Generated with Claude Code |
|
@jamjamjon Hi, I added a new model from https://github.com/53mins/CIGPose. (onnx models come from https://github.com/namas191297/cigpose-onnx). Could you check these model are applicable for usls? |
@wep21 This is amazing! CigPose looks like the new SOTA model. Thank you so much for the contribution and the PR — it’s now merged! |










What
Adds support for CIGPose, SimCC-based human pose estimation models exported to ONNX. Since the model architecture shares its I/O contract with RTMPose,
CIGPoseis a type alias ofRTMPosewith its own model configs.Model Variants
cigpose_17_lcigpose_17_l_384cigpose_133_lcigpose_133_l_384cigpose_133_x_384All variants are available in FP32 and FP16, hosted at wep21/assets cigpose release.
Changes
src/models/vision/cigpose/: model configs (CIGPose = RTMPosealias +Config::cigpose_*constructors)examples/pose-estimation/cigpose.rs+main.rs:cigposesubcommand with--scale,--is-coco,--hires,--dtype,--deviceoptionsexamples/pose-estimation/README.md,docs/model-zoo/pose.md: docsUsage
Inference results for all variants will be posted as a follow-up comment.
🤖 Generated with Claude Code