predictor

package

v0.3.1-0...-1c2b684 Latest Latest Go to latest Published: May 4, 2020 License: NCSA Imports: 18 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/rai-project/mxnet

Links

Open Source Insights

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
type ImagePredictor
type ObjectDetectionPredictor

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 ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL