predict

Jan 9, 2018

Changes in this version

+ var DefaultInputLayerName = "input"

+ var DefaultOutputLayerName = "output"

type Base

+ Options *options.Options

+ func (b Base) BatchSize() uint32

+ func (b Base) GetPredictionOptions(ctx context.Context) (*options.Options, error)

+ func (b Base) TraceLevel() tracer.Level

type ImagePredictor

+ func (p ImagePredictor) GetColorMode(defaultMode types.Mode) types.Mode

+ func (p ImagePredictor) GetFeaturesChecksum() string

+ func (p ImagePredictor) GetGraphChecksum() string

+ func (p ImagePredictor) GetInputLayerName(defaultValue string) string

+ func (p ImagePredictor) GetLayerName(typeParameters map[string]*dlframework.ModelManifest_Type_Parameter) (string, error)

+ func (p ImagePredictor) GetLayout(defaultLayout image.Layout) image.Layout

+ func (p ImagePredictor) GetOutputLayerName(defaultValue string) string

+ func (p ImagePredictor) GetPreprocessOptions(ctx context.Context) (PreprocessOptions, error)

+ func (p ImagePredictor) GetWeightsChecksum() string

type Predictor

+ GetPredictionOptions func(ctx context.Context) (*options.Options, error)

+ GetPreprocessOptions func(ctx context.Context) (PreprocessOptions, error)

type PreprocessOptions

+ Context context.Context

+ Layout image.Layout

Sep 14, 2017

Changes in this version

type Base

+ func (p Base) Info() (dlframework.FrameworkManifest, dlframework.ModelManifest, error)

type ImagePredictor

+ func (p ImagePredictor) Close() error

+ func (p ImagePredictor) PreprocessOptions(ctx context.Context) (PreprocessOptions, error)

+ func (p ImagePredictor) Reset(ctx context.Context) error

type Predictor

+ Info func() (dlframework.FrameworkManifest, dlframework.ModelManifest, error)

+ PreprocessOptions func(ctx context.Context) (PreprocessOptions, error)

+ Reset func(ctx context.Context) error

+ type PreprocessOptions struct

+ ColorMode types.Mode

+ MeanImage []float32

+ Scale float32

+ Size []int

Aug 23, 2017

Changes in this version

+ func NoMeanImageURLProcessor(ctx context.Context, url string) ([]float32, error)

+ type Base struct

+ Framework dlframework.FrameworkManifest

+ Model dlframework.ModelManifest

+ func (p Base) GetFramework() (dlframework.FrameworkManifest, error)

+ type ImagePredictor struct

+ WorkDir string

+ func (p ImagePredictor) GetFeaturesPath() string

+ func (p ImagePredictor) GetFeaturesUrl() string

+ func (p ImagePredictor) GetGraphPath() string

+ func (p ImagePredictor) GetGraphUrl() string

+ func (p ImagePredictor) GetImageDimensions() ([]int32, error)

+ func (p ImagePredictor) GetMeanImage(ctx context.Context, ...) ([]float32, error)

+ func (p ImagePredictor) GetMeanPath() string

+ func (p ImagePredictor) GetScale() (float32, error)

+ func (p ImagePredictor) GetWeightsPath() string

+ func (p ImagePredictor) GetWeightsUrl() string

+ type Predictor interface

+ Download func(ctx context.Context) error

+ GetFramework func() (dlframework.FrameworkManifest, error)

+ Load func(ctx context.Context, model dlframework.ModelManifest) (Predictor, error)

+ Predict func(ctx context.Context, data interface{}) (*dlframework.PredictionFeatures, error)

+ Preprocess func(ctx context.Context, data interface{}) (interface{}, error)