caffe

Nov 19, 2018

Changes in this version

type Predictor

+ func (p *Predictor) ReadPredictionOutput(ctx context.Context) ([]float32, error)

Jan 9, 2018

Changes in this version

type Predictor

+ func (p *Predictor) DisableProfiling() error

+ func (p *Predictor) EndProfiling() error

+ func (p *Predictor) ReadProfile() (string, error)

+ func (p *Predictor) StartProfiling(name, metadata string) error

Aug 18, 2017

Changes in this version

+ const CPUMode

+ const GPUMode

+ var BuildDate = "undefined"

+ var GitCommit = "undefined"

+ var Version = "0.3.3"

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

+ func SetUseCPU()

+ func SetUseGPU()

+ type Prediction struct

+ Index int

+ Probability float32

+ type Predictions []Prediction

+ 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 struct

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

+ func (p *Predictor) Close()

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