imagenet

Documentation

Index

• Variables
• type Classifier
  • func NewClassifier(builder func(m *godl.Model) godl.Module, width, height int) *Classifier
  • func (c *Classifier) Model() *godl.Model
  • func (c *Classifier) Predict(img image.Image) (string, float32, error)

Variables

var Labels = []string{}/* 1000 elements not displayed */

Labels are the predefined imagenet labels

Types

type Classifier

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

Classifier is an imagenet classifier

func NewClassifier

func NewClassifier(builder func(m *godl.Model) godl.Module, width, height int) *Classifier

func (*Classifier) Model

func (c *Classifier) Model() *godl.Model

func (*Classifier) Predict

func (c *Classifier) Predict(img image.Image) (string, float32, error)