Struct AddClassificationStageInterfaceProxy

struct AddClassificationStageInterfaceProxy

Interface proxy, used to insulate python bindings.

Public Static Functions

static std::shared_ptr<mrc::segment::Object<AddClassificationsStage>> init(mrc::segment::Builder &builder, const std::string &name, std::map<std::size_t, std::string> idx2label, float threshold)

Create and initialize a AddClassificationStage, and return the result.

Parameters
  • builder – : Pipeline context object reference

  • name – : Name of a stage reference

  • idx2label – : Index to classification labels map

  • threshold – : Threshold to consider true/false for each class

Returns

std::shared_ptr<mrc::segment::Object<AddClassificationsStage>>

Previous Struct MemoryDescriptor
Next Struct AddScoresStageInterfaceProxy
© Copyright 2023, NVIDIA. Last updated on Feb 2, 2024.