trtutils.impls package

Subpackages

• trtutils.impls.yolo package
  • Module contents
    • Classes
    • Functions
    • YOLO
      • YOLO.engine
      • YOLO.name
      • YOLO.input_shape
      • YOLO.dtype
      • YOLO.preprocess()
      • YOLO.postprocess()
      • YOLO.run()
      • YOLO.get_random_input()
      • YOLO.mock_run()
      • YOLO.get_detections()
      • YOLO.end2end()
    • YOLO7
    • YOLO8
    • YOLO9
    • YOLO10
    • YOLOX
    • CPUPreprocessor
      • CPUPreprocessor.warmup()
      • CPUPreprocessor.preprocess()
    • CUDAPreprocessor
      • CUDAPreprocessor.warmup()
      • CUDAPreprocessor.preprocess()
      • CUDAPreprocessor.direct_preproc()
    • ParallelYOLO
      • ParallelYOLO.models
      • ParallelYOLO.get_model()
      • ParallelYOLO.get_model_profiling()
      • ParallelYOLO.get_profiling()
      • ParallelYOLO.stop()
      • ParallelYOLO.preprocess()
      • ParallelYOLO.preprocess_model()
      • ParallelYOLO.postprocess()
      • ParallelYOLO.postprocess_model()
      • ParallelYOLO.get_detections()
      • ParallelYOLO.get_detections_model()
      • ParallelYOLO.submit()
      • ParallelYOLO.submit_model()
      • ParallelYOLO.get_random_input()
      • ParallelYOLO.mock_submit()
      • ParallelYOLO.retrieve()
      • ParallelYOLO.retrieve_model()
      • ParallelYOLO.end2end()
    • TRTPreprocessor
      • TRTPreprocessor.warmup()
      • TRTPreprocessor.preprocess()
      • TRTPreprocessor.direct_preproc()
    • get_detections()
    • postprocess()
    • preprocess()

Submodules

• trtutils.impls.common module
  • Functions
  • postprocess_efficient_nms()
  • decode_efficient_nms()
• trtutils.impls.kernels module
• trtutils.impls.onnx_models module
  • build_yolo_preproc()

Module contents

Submodule of implementations for TRTModels.

Submodules

common

Implementations which are generic to underlying models.

yolo

Implementations of YOLO object detectors.

kernels

CUDA kernels for various functions.