SegmentationModel

Objective-C

@interface SegmentationModel : Model

Swift

class SegmentationModel : Model

This class represents high-level API for segmentation models

SegmentationModel allows to set params for preprocessing input image. SegmentationModel creates net from file with trained weights and config, sets preprocessing input, runs forward pass and returns the class prediction for each pixel.

Member of Dnn

Methods

  • Create model from deep learning network.

    Declaration

    Objective-C

    - (nonnull instancetype)initWithNetwork:(nonnull Net *)network;

    Swift

    init(network: Net)

    Parameters

    network

    Net object.

  • Create segmentation model from network represented in one of the supported formats. An order of @p model and @p config arguments does not matter.

    Declaration

    Objective-C

    - (nonnull instancetype)initWithModel:(nonnull NSString *)model
                               config:(nonnull NSString *)config;

    Swift

    init(model: String, config: String)

    Parameters

    model

    Binary file contains trained weights.
    config

    Text file contains network configuration.

  • Create segmentation model from network represented in one of the supported formats. An order of @p model and @p config arguments does not matter.

    Declaration

    Objective-C

    - (nonnull instancetype)initWithModel:(nonnull NSString *)model;

    Swift

    init(model: String)

    Parameters

    model

    Binary file contains trained weights.

  • Given the @p input frame, create input blob, run net

    Declaration

    Objective-C

    - (void)segment:(nonnull Mat *)frame mask:(nonnull Mat *)mask;

    Swift

    func segment(frame: Mat, mask: Mat)

    Parameters

    mask

    Allocated class prediction for each pixel