predictor

Documentation

Index

• func NewImageClassificationPredictor(model dlframework.ModelManifest, os ...options.Option) (common.Predictor, error)
• type ImageClassificationPredictor
  • func (p *ImageClassificationPredictor) Close() error
  • func (p *ImageClassificationPredictor) Download(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) error
  • func (p *ImageClassificationPredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
  • func (p *ImageClassificationPredictor) Modality() (dlframework.Modality, error)
  • func (p *ImageClassificationPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error
  • func (p *ImageClassificationPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)
  • func (p *ImageClassificationPredictor) Reset(ctx context.Context) error

Functions

func NewImageClassificationPredictor

func NewImageClassificationPredictor(model dlframework.ModelManifest, os ...options.Option) (common.Predictor, error)

New ...

Types

type ImageClassificationPredictor

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

ImageClassificationPredictor ...

func (*ImageClassificationPredictor) Close

func (p *ImageClassificationPredictor) Close() error

Close ...

func (*ImageClassificationPredictor) Download

func (p *ImageClassificationPredictor) Download(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) error

Download ...

func (*ImageClassificationPredictor) Load

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

Load ...

func (*ImageClassificationPredictor) Modality

func (p *ImageClassificationPredictor) Modality() (dlframework.Modality, error)

func (*ImageClassificationPredictor) Predict

func (p *ImageClassificationPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

Predict ...

func (*ImageClassificationPredictor) ReadPredictedFeatures

func (p *ImageClassificationPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)

ReadPredictedFeatures ...

func (*ImageClassificationPredictor) Reset

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

Reset ...