predictor

package
v0.3.4 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Feb 8, 2021 License: NCSA Imports: 29 Imported by: 5

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	DefaultInputLayerName   = "input"
	DefaultOutputLayerNames = []string{"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) GPUMetrics 

func (b Base) GPUMetrics() 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) GetPredictionOptions 

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

func (Base) GetPreprocessOptions 

func (p Base) GetPreprocessOptions() (PreprocessOptions, error)

func (Base) GetTypeParameter 

func (p Base) GetTypeParameter(typeParameters map[string]*dlframework.ModelManifest_Type_Parameter, name string) (string, 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) Modality 

func (b Base) Modality() (dlframework.Modality, error)

func (Base) TraceLevel 

func (b Base) TraceLevel() tracer.Level

func (Base) UseGPU 

func (b Base) UseGPU() bool

type ImagePredictor 

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

func (ImagePredictor) Close 

func (p ImagePredictor) Close() error

func (ImagePredictor) CreateBoundingBoxFeatures 

func (p ImagePredictor) CreateBoundingBoxFeatures(ctx context.Context, probabilities0 interface{}, classes0 interface{}, boxes0 interface{}, labels []string) ([]dlframework.Features, error)

func (ImagePredictor) CreateBoundingBoxFeaturesFlattened 

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

func (ImagePredictor) CreateBoundingBoxFeaturesUnflattened 

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

func (ImagePredictor) CreateClassificationFeatures 

func (p ImagePredictor) CreateClassificationFeatures(ctx context.Context, probabilities0 interface{}, labels []string) ([]dlframework.Features, error)

func (ImagePredictor) CreateClassificationFeaturesFrom1D 

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

func (ImagePredictor) CreateClassificationFeaturesFrom2D 

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

func (ImagePredictor) CreateImageFeatures 

func (p ImagePredictor) CreateImageFeatures(ctx context.Context, images [][][][]float32) ([]dlframework.Features, error)

func (ImagePredictor) CreateInstanceSegmentFeatures 

func (p ImagePredictor) CreateInstanceSegmentFeatures(ctx context.Context, probabilities [][]float32, classes [][]float32, boxes [][][]float32, masks [][][][]float32, labels []string) ([]dlframework.Features, error)

func (ImagePredictor) CreateRawImageFeatures 

func (p ImagePredictor) CreateRawImageFeatures(ctx context.Context, images [][][][]float32) ([]dlframework.Features, error)

func (ImagePredictor) CreateSemanticSegmentFeatures 

func (p ImagePredictor) CreateSemanticSegmentFeatures(ctx context.Context, masks0 interface{}, labels []string) ([]dlframework.Features, error)

func (ImagePredictor) CreateSemanticSegmentFeaturesFrom2D 

func (p ImagePredictor) CreateSemanticSegmentFeaturesFrom2D(ctx context.Context, masks [][][]int64, labels []string) ([]dlframework.Features, error)

func (ImagePredictor) GetBackgroundIndex added in v0.3.2 

func (p ImagePredictor) GetBackgroundIndex() (bool, int)

func (ImagePredictor) GetBoxIndex added in v0.3.2 

func (p ImagePredictor) GetBoxIndex() []int

func (ImagePredictor) GetBoxScaling added in v0.3.3 

func (p ImagePredictor) GetBoxScaling() (float32, float32)

func (ImagePredictor) GetColorMode 

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

func (ImagePredictor) GetCropMethod 

func (p ImagePredictor) GetCropMethod(defaultMethod cutter.AnchorMode) cutter.AnchorMode

func (ImagePredictor) GetCropRatio 

func (p ImagePredictor) GetCropRatio(defaultCropRatio float32) float32

func (ImagePredictor) GetFeatureType 

func (p ImagePredictor) GetFeatureType() dlframework.FeatureType

func (ImagePredictor) GetFeaturesChecksum 

func (p ImagePredictor) GetFeaturesChecksum() string

func (ImagePredictor) GetFeaturesPath 

func (p ImagePredictor) GetFeaturesPath() string

func (ImagePredictor) GetInputDimensions 

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

func (ImagePredictor) GetInputLayerName 

func (p ImagePredictor) GetInputLayerName(layer string) (string, error)

func (ImagePredictor) GetKeepAspectRatio 

func (p ImagePredictor) GetKeepAspectRatio() (bool, error)

func (ImagePredictor) GetLabels 

func (p ImagePredictor) GetLabels() ([]string, error)

func (ImagePredictor) GetLayout 

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

func (ImagePredictor) GetMaxDimension 

func (p ImagePredictor) GetMaxDimension() (int, error)

func (ImagePredictor) GetMeanImage 

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

func (ImagePredictor) GetMeanPath 

func (p ImagePredictor) GetMeanPath() string

func (ImagePredictor) GetOutputLayerIndex 

func (p ImagePredictor) GetOutputLayerIndex(layer string) (int, error)

func (ImagePredictor) GetPreprocessOptions 

func (p ImagePredictor) GetPreprocessOptions() (PreprocessOptions, error)

func (ImagePredictor) GetProbabilitiesTransform 

func (p ImagePredictor) GetProbabilitiesTransform() 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)
	// Gets predictor's Modality
	Modality() (dlframework.Modality, 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() (*options.Options, error)
	// Returns the preprocess options
	GetPreprocessOptions() (PreprocessOptions, error)
	// Returns the handle to features
	Predict(ctx context.Context, data interface{}, 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 {
	ElementType     string
	MeanImage       []float32
	Scale           []float32
	Dims            []int
	MaxDimension    *int
	KeepAspectRatio *bool
	ColorMode       types.Mode
	Layout          raiimage.Layout
	CropMethod      cutter.AnchorMode
	CropRatio       float32
}

type RawPredictor 

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

func (RawPredictor) Close 

func (p RawPredictor) Close() error

func (RawPredictor) GetInputDataByIdx 

func (p RawPredictor) GetInputDataByIdx(idx int) ([]interface{}, error)

func (RawPredictor) GetInputParams 

func (p RawPredictor) GetInputParams(name string) ([]string, error)

func (RawPredictor) GetInputParamsByIdx 

func (p RawPredictor) GetInputParamsByIdx(idx int, name string) (string, error)

func (RawPredictor) GetInputURLs 

func (p RawPredictor) GetInputURLs() ([]string, error)

func (RawPredictor) GetOutputLayerName 

func (p RawPredictor) GetOutputLayerName(layer string) (string, error)

func (RawPredictor) ReadPredictedFeatures 

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

func (RawPredictor) Reset 

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

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.