OpenmindAGI/person-following-models

Person Following — Model Files

Models consumed by person_following.gesture and the YOLO detection pipeline. Mix of:

• ONNX sources — portable, compile locally to TRT on the target device.
• TRT engines — GPU-specific prebuilt for faster cold start. Falls back to ONNX-compile if the engine fails compatibility check.
• PyTorch checkpoints — loaded directly by Python code.

Files

Name	Type	Notes
rtmw.onnx	ONNX	RTMW pose, 133 keypoints, 384x288
rtmw_l_fp16_dynbatch.engine	TRT engine	FP16, dynamic batch 1-6, built on Thor
action_recog_stgcn.pth	PyTorch	7-class HRI gesture (OccAug v3)
yolo11n.engine	TRT engine	Person detection (built on Thor)
yolo11s-seg.engine	TRT engine	Person segmentation (built on Thor)

Engine portability caveat

TRT engines are not portable across GPU architectures or TRT versions. The engines here were built for NVIDIA Jetson Thor / TRT 10.16. On other hardware, ModelManager will detect the incompatibility, discard the downloaded engine, and rebuild from ONNX.