cls

Documentation

Index

• Constants
• func ColorFormatBGR(r, g, b float32) (first, second, third float32)
• func ColorFormatRGB(r, g, b float32) (first, second, third float32)
• type Classifier
  • func NewClassifier(conf *Config, sessionConf *SessionConfig) (*Classifier, error)
  • func (c *Classifier) Close()
  • func (c *Classifier) Run(images []image.Image) ([]*Output, error)
  • func (c *Classifier) RunRaw(images []image.Image) ([][][]float32, error)
• type ColorFormatFunc
• type Config
• type CoreMLConfig
• type CudaConfig
• type DirectMLConfig
• type LabelProb
  • func (l *LabelProb) Get() (string, float32)
• type OpenVINOConfig
• type Output
  • func (o *Output) Labels() [][]*LabelProb
  • func (o *Output) Raw() [][]float32
  • func (o *Output) TopLabels() []*LabelProb
• type ProcessFloatImageFunc
  • func Normalize(mean, std float32) ProcessFloatImageFunc
  • func Rescale() ProcessFloatImageFunc
• type ProcessImageFunc
  • func ResizeImage(size int, interp resize.InterpolationFunction) ProcessImageFunc
  • func ResizeWithPadding(size int, padColor color.Color, interp resize.InterpolationFunction) ProcessImageFunc
• type SessionConfig
• type Shape
• type TensorRTConfig

Constants

const (
	BCHW = iota
	BHWC
)

Functions

func ColorFormatBGR

func ColorFormatBGR(r, g, b float32) (first, second, third float32)

func ColorFormatRGB

func ColorFormatRGB(r, g, b float32) (first, second, third float32)

Types

type Classifier

type Classifier struct {
	// contains filtered or unexported fields
}

func NewClassifier

func NewClassifier(conf *Config, sessionConf *SessionConfig) (*Classifier, error)

func (*Classifier) Close

func (c *Classifier) Close()

func (*Classifier) Run

func (c *Classifier) Run(images []image.Image) ([]*Output, error)

func (*Classifier) RunRaw

func (c *Classifier) RunRaw(images []image.Image) ([][][]float32, error)

type ColorFormatFunc

type ColorFormatFunc func(r, g, b float32) (first, second, third float32)

type Config

type Config struct {
	ModelPath              string
	Width, Height          int
	OutputLabels           map[int][]string
	Shape                  Shape
	ColorFormatFunc        ColorFormatFunc
	ProcessImageFuncs      []ProcessImageFunc
	ProcessFloatImageFuncs []ProcessFloatImageFunc
}

type CoreMLConfig

type CoreMLConfig struct {
	Enabled bool
	//https://github.com/microsoft/onnxruntime/blob/291a5352b27ded5714e5748b381f2efb88f28fb9/include/onnxruntime/core/providers/coreml/coreml_provider_factory.h#L12
	Flags uint32
}

type CudaConfig

type CudaConfig struct {
	Enabled bool
	// https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#configuration-options
	Options map[string]string
}

type DirectMLConfig

type DirectMLConfig struct {
	Enabled  bool
	DeviceID int
}

type LabelProb

type LabelProb struct {
	// contains filtered or unexported fields
}

func (*LabelProb) Get

func (l *LabelProb) Get() (string, float32)

type OpenVINOConfig

type OpenVINOConfig struct {
	Enabled bool
	//https://onnxruntime.ai/docs/execution-providers/OpenVINO-ExecutionProvider.html#summary-of-options
	Options map[string]string
}

type Output

type Output struct {
	// contains filtered or unexported fields
}

func (*Output) Labels

func (o *Output) Labels() [][]*LabelProb

func (*Output) Raw

func (o *Output) Raw() [][]float32

func (*Output) TopLabels

func (o *Output) TopLabels() []*LabelProb

type ProcessFloatImageFunc

type ProcessFloatImageFunc func(rgb float32) float32

func Normalize

func Normalize(mean, std float32) ProcessFloatImageFunc

func Rescale

func Rescale() ProcessFloatImageFunc

type ProcessImageFunc

type ProcessImageFunc func(img image.Image) image.Image

func ResizeImage

func ResizeImage(size int, interp resize.InterpolationFunction) ProcessImageFunc

func ResizeWithPadding

func ResizeWithPadding(size int, padColor color.Color, interp resize.InterpolationFunction) ProcessImageFunc

type SessionConfig

type SessionConfig struct {
	Cuda        CudaConfig
	TensortRT   TensorRTConfig
	CoreML      CoreMLConfig
	DirectML    DirectMLConfig
	CpuMemArena *bool
	MemPattern  *bool

	InterOpNumThreads int
	IntraOpNumThreads int
}

type Shape

type Shape int

type TensorRTConfig

type TensorRTConfig struct {
	Enabled bool
	// https://onnxruntime.ai/docs/execution-providers/TensorRT-ExecutionProvider.html#execution-provider-options
	Options map[string]string
}