Contents

init(modelPath:outputName:)

Creates a custom feature extractor given a model file and an optional output layer name.

Declaration

init(modelPath: URL, outputName: String? = nil)

Parameters

  • modelPath:

    The location of the neural network .mlmodel which contains the feature extractor.

  • outputName:

    The name of a feature extraction layer within the model has one output type of Mlmultiarray. Set this value to nil if the model, as a whole, accepts an input image and has exactly one output of type Mlmultiarray.