predict

Documentation

Index

• Variables
• func New(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
• type ImagePredictor
  • func (p *ImagePredictor) Close() error
  • func (p *ImagePredictor) GetPreprocessOptions(ctx context.Context) (common.PreprocessOptions, error)
  • func (p *ImagePredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
  • func (p *ImagePredictor) Predict(ctx context.Context, data [][]float32, opts ...options.Option) ([]dlframework.Features, error)
  • func (p *ImagePredictor) Reset(ctx context.Context) error

Variables

var (
	DefaultOutputLayerName = "z"
	EnableFrameworkProfile = true
)

Functions

func New

func New(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

New ...

Types

type ImagePredictor

type ImagePredictor struct {
	common.ImagePredictor
	// contains filtered or unexported fields
}

ImagePredictor ...

func (*ImagePredictor) Close

func (p *ImagePredictor) Close() error

Close ...

func (*ImagePredictor) GetPreprocessOptions

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

GetPreprocessOptions ...

func (*ImagePredictor) Load

func (p *ImagePredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

Load ...

func (*ImagePredictor) Predict

func (p *ImagePredictor) Predict(ctx context.Context, data [][]float32, opts ...options.Option) ([]dlframework.Features, error)

Predict ...

func (*ImagePredictor) Reset

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

Reset ...