tensor

Documentation

Index

• type Config
  • func NewConfig(configPath string) *Config
• type Tensor
  • func NewTensor(name string) *Tensor
  • func (t *Tensor) Run(input <-chan videosource.ProcessedImage) <-chan videosource.ProcessedImage
  • func (t *Tensor) SetConfig(config *Config)

Types

type Config

type Config struct {
	Skip                    bool     `yaml:"skip,omitempty"`
	ForceCpu                bool     `yaml:"forceCpu,omitempty"`
	Padding                 int      `yaml:"padding,omitempty"`
	ModelFile               string   `yaml:"modelFile,omitempty"`
	ConfigFile              string   `yaml:"configFile,omitempty"`
	DescFile                string   `yaml:"descFile,omitempty"`
	ScaleWidth              int      `yaml:"scaleWidth,omitempty"`
	MinConfidencePercentage int      `yaml:"minConfidencePercentage,omitempty"`
	MinMotionFrames         int      `yaml:"minMotionFrames,omitempty"`
	MinPercentage           int      `yaml:"minPercentage,omitempty"`
	MaxPercentage           int      `yaml:"maxPercentage,omitempty"`
	MinOverlapPercentage    int      `yaml:"minOverlapPercentage,omitempty"`
	SameOverlapPercentage   int      `yaml:"sameOverlapPercentage,omitempty"`
	AllowedList             []string `yaml:"allowedList,omitempty"`
	HighlightColor          string   `yaml:"highlightColor,omitempty"`
	HighlightThickness      int      `yaml:"highlightThickness,omitempty"`
}

Config contains the parameters for tensor detection

func NewConfig

func NewConfig(configPath string) *Config

NewConfig creates a new Config

type Tensor

type Tensor struct {
	Name string
	Skip bool
	// contains filtered or unexported fields
}

Tensor detects objects within images

func NewTensor

func NewTensor(name string) *Tensor

NewTensor creates a new Tensor

func (*Tensor) Run

func (t *Tensor) Run(input <-chan videosource.ProcessedImage) <-chan videosource.ProcessedImage

Run starts the tensor detection process

func (*Tensor) SetConfig

func (t *Tensor) SetConfig(config *Config)

SetConfig on tensor