Segmentation Model

class SegmentationModel : public ImageModel

Public Functions

SegmentationModel(std::shared_ptr<ov::Model> &model, const ov::AnyMap &configuration)

SegmentationModel(std::shared_ptr<InferenceAdapter> &adapter, const ov::AnyMap &configuration = {})

virtual std::unique_ptr<ResultBase> postprocess(InferenceResult &infResult) override

virtual std::unique_ptr<ImageResult> infer(const ImageInputData &inputData)

virtual std::vector<std::unique_ptr<ImageResult>> inferBatch(const std::vector<ImageInputData> &inputImgs)

std::vector<Contour> getContours(const ImageResultWithSoftPrediction &imageResult)

Public Static Functions

static std::unique_ptr<SegmentationModel> create_model(const std::string &modelFile, const ov::AnyMap &configuration = {}, bool preload = true, const std::string &device = "AUTO")

static std::unique_ptr<SegmentationModel> create_model(std::shared_ptr<InferenceAdapter> &adapter)

Public Static Attributes

static std::string ModelType = "Segmentation"