pesi
/

rtmo

Object Detection

Pose Estimation

Model card Files Files and versions Community

rtmo / rtmo_gpu.py

Commit History

Unify batched and non-batched versions

42892de

Luigi commited on May 17

Support Inference over batch with TensorRT Engine Model

0cdc9a7

Luigi commited on May 17

Use ONNXruntime instead of ONNX.checker.check_model to detect ONNX model

57a8c6b

Luigi commited on May 17

Resize in keeping aspect ratio in visualization

35b2a45

Luigi commited on May 16

fix postprocess_batch method in RTMO_GPU_Batch class

6008f96

Luigi commited on May 16

Add model format (ONNX/Engine) & input size detection (based on file header, not on filename) for RTMO

58a44cf

Luigi commited on May 15

Add support TensorRT engine support for RTMO

bbf20b6

Luigi commited on Apr 19

Load ./libmmdeploy_tensorrt_ops.so for TensorRT EP if available

1c5dc58

Luigi commited on Mar 19

Add option --yolo_nas_pose, used to read YOLO NAS Pose model instead of RTMO

1f0f5d8

Luigi commited on Mar 8

Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider

b75f05d

Luigi commited on Mar 7

Add new class 'RTMO_GPU_Batch' that can perform inference on batch of images

8072759

Luigi commited on Mar 5

Optimization for CUDAExecutionProvider

43a489d

Luigi commited on Mar 5

RTMO ONNX Models from MMPOSE range from Tiny to Large

7ad7e4d
verified

Luigi commited on Mar 1