caffe

README

GO Bindings for Caffe Prediction

Documentation

Index

• Constants
• Variables
• func ReadBlob(protoFileName string) (*caffeproto.BlobProto, error)
• func SetUseCPU()
• func SetUseGPU()
• type Prediction
• type Predictions
  • func (p Predictions) Len() int
  • func (p Predictions) Less(i, j int) bool
  • func (p Predictions) Sort()
  • func (p Predictions) Swap(i, j int)
• type Predictor
  • func New(modelFile, trainFile string) (*Predictor, error)
  • func (p *Predictor) Close()
  • func (p *Predictor) Predict(imageData []float32) (Predictions, error)

Constants

const (
	CPUMode = 0
	GPUMode = 1
)

Variables

var (
	Version   = "0.3.5"
	BuildDate = "undefined"
	GitCommit = "undefined"
)

Functions

func ReadBlob

func ReadBlob(protoFileName string) (*caffeproto.BlobProto, error)

func SetUseCPU

func SetUseCPU()

func SetUseGPU

func SetUseGPU()

Types

type Prediction

type Prediction struct {
	Index       int     `json:"index"`
	Probability float32 `json:"probability"`
}

type Predictions

type Predictions []Prediction

func (Predictions) Len

func (p Predictions) Len() int

Len is the number of elements in the collection.

func (Predictions) Less

func (p Predictions) Less(i, j int) bool

Less reports whether the element with index i should sort before the element with index j.

func (Predictions) Sort

func (p Predictions) Sort()

func (Predictions) Swap

func (p Predictions) Swap(i, j int)

Swap swaps the elements with indexes i and j.

type Predictor

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

func New

func New(modelFile, trainFile string) (*Predictor, error)

func (*Predictor) Close

func (p *Predictor) Close()

func (*Predictor) Predict

func (p *Predictor) Predict(imageData []float32) (Predictions, error)