classifier

Documentation

Index

• Variables
• type Classifier
  • func NewClassifier(graphPath, labelPath string) (*Classifier, error)
  • func NewClassifierWithConfig(graphPath, labelPath string, config Config) (*Classifier, error)
  • func (c *Classifier) ClassifyBytes(img []byte, ext string) ([]Prediction, error)
  • func (c *Classifier) ClassifyImage(img image.Image) ([]Prediction, error)
  • func (c *Classifier) ClassifyMultipart(fh *multipart.FileHeader) ([]Prediction, error)
  • func (c *Classifier) Close() error
• type Config
• type Prediction

Variables

var DefaultConfig = Config{
	Height:         299,
	Width:          299,
	Mean:           0,
	Scale:          255,
	InputLayer:     "Placeholder",
	OutputLayer:    "final_result",
	NumPredictions: 5,
}

DefaultConfig is used in the case where a config is not defined by the user

Types

type Classifier

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

Classifier is a struct used for classifying images

func NewClassifier

func NewClassifier(graphPath, labelPath string) (*Classifier, error)

NewClassifier creates a new Classifier using the default config

func NewClassifierWithConfig

func NewClassifierWithConfig(graphPath, labelPath string,
	config Config) (*Classifier, error)

NewClassifierWithConfig creates a new image Classifier for processing image predictions

func (*Classifier) ClassifyBytes

func (c *Classifier) ClassifyBytes(img []byte, ext string) ([]Prediction, error)

ClassifyBytes classifies the passed images bytes and returns the predictions as a slice of Predictions

func (*Classifier) ClassifyImage

func (c *Classifier) ClassifyImage(img image.Image) ([]Prediction, error)

ClassifyImage takes an image Image, writes it to a bytes Buffer, performs a classification and returns a slice of predictions

func (*Classifier) ClassifyMultipart

func (c *Classifier) ClassifyMultipart(fh *multipart.FileHeader) ([]Prediction, error)

ClassifyMultipart takes a multipart Fileheader, performs a classification and returns a slice of predictions

func (*Classifier) Close

func (c *Classifier) Close() error

Close closes the Classifier by closing all it's closers ;)

type Config

type Config struct {
	Height, Width           int32
	Mean, Scale             float32
	InputLayer, OutputLayer string
	NumPredictions          int
}

Config is a struct used for configuring the classifier

type Prediction

type Prediction struct {
	Label       string  `json:"label"`
	Probability float32 `json:"probability"`
}

Prediction is a struct containing a class label and the probability of the classified image being the cooresponding label