rtmo / rtmo_gpu.py

Commit History

  1. adjust visualisation style, allow draw_bbox() to show person id

    a9e21bc

  1. Add method to free buffers

    b79a8af

  1. Implement pycuda backend for inference with TensorRT engine⁠

    889281f

  1. Add 'get_model_format_and_input_shape' helper function

    Luigi
    2f2f685
  2. Return and show bounding box confidence

    634d4ff

  1. Deal with multiple cameras

    Luigi
    98d66d4

  1. Show bounding box on screen too

    Luigi
    0bf1eb7
  2. Internalize input frame buffer

    Luigi
    846e714
  3. Unify batched and non-batched versions

    Luigi
    42892de
  4. Support Inference over batch with TensorRT Engine Model

    0cdc9a7
  5. Use ONNXruntime instead of ONNX.checker.check_model to detect ONNX model⁠

    57a8c6b

  1. Resize in keeping aspect ratio in visualization

    35b2a45
  2. fix postprocess_batch method in RTMO_GPU_Batch class

    6008f96

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

    58a44cf

  1. Add support TensorRT engine support for RTMO

    bbf20b6

  1. Load ./libmmdeploy_tensorrt_ops.so for TensorRT EP if available

    1c5dc58

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

    1f0f5d8

  1. Make ONNX models compatible with ONNXruntime's TensorrtExecutionProvider⁠

    b75f05d

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

    8072759
  2. Optimization for CUDAExecutionProvider⁠

    43a489d

  1. RTMO ONNX Models from MMPOSE range from Tiny to Large

    Verified signature
    7ad7e4d