Create the neural network layers for a single stage.

Get model and engine names and paths for FoundationPose tests. Returns ------- dict Dictionary containing model and engine n

Generate both Refine and Score TensorRT engine files for FoundationPose tests. This function should be called at the beginning of generate_t

Generate TensorRT engine file from ONNX model using trtexec. Parameters ---------- engine_path : str Path where the engine f

Returns 3x9 matrix of the 3d coordinates of cuboid corners and center

Returns pixel coordinate (row, col) of maxima in single-channel image

Returns pixel mask for local maximums in single - channel image src

Get the path to trtexec executable. Returns ------- str Path to trtexec executable, checking PATH first, then default locati

Convert pre-trained Pytorch DOPE model to given format for TensorRT and/or Triton to use. This is tested for the pre-train weights linked fr