mlvision

Documentation

Overview

Package mlvision uses an underlying model from the ML model service as a vision model, and wraps the ML model with the vision service methods.

Index

• Constants
• type MLModelConfig

Constants

const (
	// UInt8 is one of the possible input/output types for tensors.
	UInt8 = "uint8"
	// Float32 is one of the possible input/output types for tensors.
	Float32 = "float32"
	// DefaultOutTensorName is the prefix key given to output tensors in the map
	// if there is no metadata. (output0, output1, etc.)
	DefaultOutTensorName = "output"
)

Types

type MLModelConfig

type MLModelConfig struct {
	resource.TriviallyValidateConfig
	ModelName string `json:"mlmodel_name"`
}

MLModelConfig specifies the parameters needed to turn an ML model into a vision Model.