Documentation
¶
Index ¶
- Variables
- func New(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
- type ImagePredictor
- func (p *ImagePredictor) Close() error
- func (p *ImagePredictor) Download(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) 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) error
- func (p *ImagePredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)
- func (p *ImagePredictor) Reset(ctx context.Context) error
Constants ¶
This section is empty.
Variables ¶
View Source
var (
DefaultInputLayerName = "data"
)
Functions ¶
func New ¶
func New(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
Types ¶
type ImagePredictor ¶
type ImagePredictor struct {
common.ImagePredictor
// contains filtered or unexported fields
}
func (*ImagePredictor) Close ¶
func (p *ImagePredictor) Close() error
func (*ImagePredictor) Download ¶
func (p *ImagePredictor) Download(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) error
func (*ImagePredictor) GetPreprocessOptions ¶ added in v0.2.21
func (p *ImagePredictor) GetPreprocessOptions(ctx context.Context) (common.PreprocessOptions, error)
func (*ImagePredictor) Load ¶ added in v0.2.15
func (p *ImagePredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
func (*ImagePredictor) ReadPredictedFeatures ¶ added in v0.3.0
func (p *ImagePredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)
ReadPredictedFeatures ...
Source Files
Click to show internal directories.
Click to hide internal directories.