Class TransformBase

Derived Type

class TransformBase

Base Transform Class.

Subclassed by holoscan::inference::GenerateBoxes

Public Functions

inline virtual InferStatus execute(const std::map<std::string, void*> &indata, const std::map<std::string, std::vector<int>> &indim, DataMap &processed_data, DimType &processed_dims)

Does the transform execution.

Parameters
  • indataMap with key as tensor name and value as raw data buffer

  • indimMap with key as tensor name and value as dimension of the input tensor

  • processed_data – Output data map, that will be populated

  • processed_dims – Dimension of the output tensor, is populated during the processing

Returns

InferStatus

inline virtual InferStatus initialize(const std::vector<std::string> &input_tensors)

Previous Class TorchInfer
Next Class TrtInfer
© Copyright 2022-2023, NVIDIA. Last updated on Feb 16, 2024.