caffe

package module

v0.3.13 Latest Latest Go to latest Published: Jun 29, 2018 License: NCSA Imports: 20 Imported by: 0

Details

Rendered for

View Source

const (
	CPUMode = 0
	GPUMode = 1
)

View Source

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

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

func SetUseCPU()

func SetUseGPU()

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

type Predictions []Prediction

func (p Predictions) Len() int

Len is the number of elements in the collection.

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 (p Predictions) Sort()

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

Swap swaps the elements with indexes i and j.

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

func New(opts ...options.Option) (*Predictor, error)

func (p *Predictor) Close()

func (p *Predictor) DisableProfiling() error

func (p *Predictor) EndProfiling() error

func (p *Predictor) Predict(ctx context.Context, data []float32) (Predictions, error)

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

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

Path	Synopsis
examples
batch command
single command
proto Package proto is a generated protocol buffer package.	Package proto is a generated protocol buffer package.