Documentation
¶
Index ¶
- Variables
- func NewImageClassificationPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
- func NewObjectDetectionPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
- type ImageClassificationPredictor
- func (self *ImageClassificationPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, ...) (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)
- type ImagePredictor
- type ObjectDetectionPredictor
- func (self *ObjectDetectionPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, ...) (common.Predictor, error)
- func (p ObjectDetectionPredictor) Modality() (dlframework.Modality, error)
- func (p *ObjectDetectionPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error
- func (p *ObjectDetectionPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)
Constants ¶
This section is empty.
Variables ¶
View Source
var (
DefaultInputLayerName = "data"
)
Functions ¶
func NewImageClassificationPredictor ¶
func NewImageClassificationPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
func NewObjectDetectionPredictor ¶
func NewObjectDetectionPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
Types ¶
type ImageClassificationPredictor ¶
type ImageClassificationPredictor struct {
*ImagePredictor
// contains filtered or unexported fields
}
func (*ImageClassificationPredictor) Load ¶
func (self *ImageClassificationPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
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 ...
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
Download ...
func (*ImagePredictor) Load ¶
func (p *ImagePredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (*ImagePredictor, error)
type ObjectDetectionPredictor ¶
type ObjectDetectionPredictor struct {
*ImagePredictor
// contains filtered or unexported fields
}
func (*ObjectDetectionPredictor) Load ¶
func (self *ObjectDetectionPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
func (ObjectDetectionPredictor) Modality ¶
func (p ObjectDetectionPredictor) Modality() (dlframework.Modality, error)
func (*ObjectDetectionPredictor) Predict ¶
func (p *ObjectDetectionPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error
Predict ...
func (*ObjectDetectionPredictor) ReadPredictedFeatures ¶
func (p *ObjectDetectionPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)
ReadPredictedFeatures ...
Source Files
Click to show internal directories.
Click to hide internal directories.