predict

package

v0.2.23-0...-54c2d5c Latest Latest Go to latest Published: Nov 26, 2018 License: NCSA Imports: 16 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/rai-project/dlframework

Links

Open Source Insights

Documentation ¶

Constants ¶

This section is empty.

Variables ¶

View Source

var (
	DefaultInputLayerName  = "input"
	DefaultOutputLayerName = "output"
)

Functions ¶

This section is empty.

Types ¶

type Base ¶

type Base struct {
	Framework dlframework.FrameworkManifest
	Model     dlframework.ModelManifest
	WorkDir   string
	Options   *options.Options
}

func (Base) BatchSize ¶

func (b Base) BatchSize() int

func (Base) FeatureLimit ¶

func (b Base) FeatureLimit() int

func (Base) GetFeaturesChecksum ¶

func (p Base) GetFeaturesChecksum() string

func (Base) GetFeaturesPath ¶

func (p Base) GetFeaturesPath() string

func (Base) GetFeaturesUrl ¶

func (p Base) GetFeaturesUrl() string

func (Base) GetGraphChecksum ¶

func (p Base) GetGraphChecksum() string

func (Base) GetGraphPath ¶

func (p Base) GetGraphPath() string

func (Base) GetGraphUrl ¶

func (p Base) GetGraphUrl() string

func (Base) GetInputLayerName ¶

func (p Base) GetInputLayerName(defaultValue string) string

func (Base) GetLayerName ¶

func (p Base) GetLayerName(typeParameters map[string]*dlframework.ModelManifest_Type_Parameter) (string, error)

func (Base) GetOutputLayerName ¶

func (p Base) GetOutputLayerName(defaultValue string) string

func (Base) GetPredictionOptions ¶

func (b Base) GetPredictionOptions(ctx context.Context) (*options.Options, error)

func (Base) GetPreprocessOptions ¶

func (p Base) GetPreprocessOptions(ctx context.Context) (PreprocessOptions, error)

func (Base) GetWeightsChecksum ¶

func (p Base) GetWeightsChecksum() string

func (Base) GetWeightsPath ¶

func (p Base) GetWeightsPath() string

func (Base) GetWeightsUrl ¶

func (p Base) GetWeightsUrl() string

func (Base) Info ¶

func (b Base) Info() (dlframework.FrameworkManifest, dlframework.ModelManifest, error)

func (Base) TraceLevel ¶

func (b Base) TraceLevel() tracer.Level

type ImagePredictor ¶

type ImagePredictor struct {
	Base
	Metadata map[string]interface{}
}

func (ImagePredictor) Close ¶

func (p ImagePredictor) Close() error

func (ImagePredictor) CreatePredictedFeatures ¶

func (p ImagePredictor) CreatePredictedFeatures(ctx context.Context, probabilities []float32, labels []string) ([]dlframework.Features, error)

ReadPredictedFeatures ...

func (ImagePredictor) GetColorMode ¶

func (p ImagePredictor) GetColorMode(defaultMode types.Mode) types.Mode

func (ImagePredictor) GetImageDimensions ¶

func (p ImagePredictor) GetImageDimensions() ([]int, error)

func (ImagePredictor) GetLayout ¶

func (p ImagePredictor) GetLayout(defaultLayout image.Layout) image.Layout

func (ImagePredictor) GetMeanImage ¶

func (p ImagePredictor) GetMeanImage() ([]float32, error)

func (ImagePredictor) GetMeanPath ¶

func (p ImagePredictor) GetMeanPath() string

func (ImagePredictor) GetScale ¶

func (p ImagePredictor) GetScale() (float32, error)

func (ImagePredictor) Reset ¶

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

type Predictor ¶

type Predictor interface {
	// Gets framework and model manifests
	Info() (dlframework.FrameworkManifest, dlframework.ModelManifest, error)
	// Downloads model from manifest
	Download(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) error
	// Load model from manifest
	Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (Predictor, error)
	// Returns the prediction options
	GetPredictionOptions(ctx context.Context) (*options.Options, error)
	// Returns the preprocess options
	GetPreprocessOptions(ctx context.Context) (PreprocessOptions, error)
	// Returns the handle to features
	Predict(ctx context.Context, data [][]float32, opts ...options.Option) error
	// Returns the features
	ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)
	// Clears the internal state of a predictor
	Reset(ctx context.Context) error

	io.Closer
}

type PreprocessOptions ¶

type PreprocessOptions struct {
	Context   context.Context
	MeanImage []float32
	Size      []int
	Scale     float32
	ColorMode types.Mode
	Layout    image.Layout
}

Source Files ¶

View all Source files

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