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
  • func (conf *MLModelConfig) Validate(path string) ([]string, error)

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 {
	ModelName string `json:"mlmodel_name"`
}

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

func (*MLModelConfig) Validate

func (conf *MLModelConfig) Validate(path string) ([]string, error)

Validate will add the ModelName as an implicit dependency to the robot.