predictor

Mar 24, 2022

Changes in this version

type GeneralPredictor

+ func (p *GeneralPredictor) SetDesiredOutput(modality dlframework.Modality)

Nov 14, 2021

Changes in this version

+ func NewGeneralPredictor(model dlframework.ModelManifest, os ...options.Option) (common.Predictor, error)

+ type GeneralPredictor struct

+ func (p *GeneralPredictor) Close() error

+ func (p *GeneralPredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func (p *GeneralPredictor) Modality() (dlframework.Modality, error)

+ func (p *GeneralPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

+ func (p *GeneralPredictor) ReadPredictedFeaturesAsMap(ctx context.Context) (map[string]interface{}, error)

+ func (p *GeneralPredictor) Reset(ctx context.Context) error

Feb 1, 2021

Changes in this version

+ func ElementTypeSliceToTensor(data [][]ElementType, shape []int64) (*tf.Tensor, error)

+ func Float32SliceToTensor(data [][]float32, shape []int64) (*tf.Tensor, error)

+ func Float64SliceToTensor(data [][]float64, shape []int64) (*tf.Tensor, error)

+ func Int16SliceToTensor(data [][]int16, shape []int64) (*tf.Tensor, error)

+ func Int32SliceToTensor(data [][]int32, shape []int64) (*tf.Tensor, error)

+ func Int64SliceToTensor(data [][]int64, shape []int64) (*tf.Tensor, error)

+ func Int8SliceToTensor(data [][]int8, shape []int64) (*tf.Tensor, error)

+ func NewImageClassificationPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func NewImageEnhancementPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func NewInstanceSegmentationPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func NewObjectDetectionPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func NewRawPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func NewSemanticSegmentationPredictor(model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func Uint16SliceToTensor(data [][]uint16, shape []int64) (*tf.Tensor, error)

+ func Uint32SliceToTensor(data [][]uint32, shape []int64) (*tf.Tensor, error)

+ func Uint64SliceToTensor(data [][]uint64, shape []int64) (*tf.Tensor, error)

+ func Uint8SliceToTensor(data [][]uint8, shape []int64) (*tf.Tensor, error)

+ type Device struct

+ MemoryLimitBytes int64

+ Name string

+ Type string

+ func (d Device) String() string

+ type ElementType generic.Type

+ type Graph = tf.Graph

+ type ImageClassificationPredictor struct

+ func (p *ImageClassificationPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

+ func (p *ImageClassificationPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)

+ func (p ImageClassificationPredictor) Modality() (dlframework.Modality, error)

+ func (self *ImageClassificationPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, ...) (common.Predictor, error)

+ type ImageEnhancementPredictor struct

+ func (p *ImageEnhancementPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

+ func (p *ImageEnhancementPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)

+ func (p ImageEnhancementPredictor) Modality() (dlframework.Modality, error)

+ func (self *ImageEnhancementPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, ...) (common.Predictor, error)

+ type ImagePredictor struct

+ func (p *ImagePredictor) Close() error

+ func (p *ImagePredictor) Download(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) error

+ func (p *ImagePredictor) GetInputLayerName(reader io.Reader, layer string) (string, error)

+ func (p *ImagePredictor) GetOutputLayerName(reader io.Reader, layer string) (string, error)

+ func (p *ImagePredictor) GetOutputParameter(key string) string

+ func (p *ImagePredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (*ImagePredictor, error)

+ type InstanceSegmentationPredictor struct

+ func (p *InstanceSegmentationPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

+ func (p *InstanceSegmentationPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)

+ func (p InstanceSegmentationPredictor) Modality() (dlframework.Modality, error)

+ func (self *InstanceSegmentationPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, ...) (common.Predictor, error)

+ type ObjectDetectionPredictor struct

+ func (p *ObjectDetectionPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

+ func (p *ObjectDetectionPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)

+ func (p ObjectDetectionPredictor) Modality() (dlframework.Modality, error)

+ func (self *ObjectDetectionPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, ...) (common.Predictor, error)

+ type Operation = tf.Operation

+ type Output = tf.Output

+ type RawPredictor struct

+ func (p *RawPredictor) Close() error

+ func (p *RawPredictor) Download(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) error

+ func (p *RawPredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

+ func (p *RawPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

+ func (p RawPredictor) Modality() (dlframework.Modality, error)

+ type SemanticSegmentationPredictor struct

+ func (p *SemanticSegmentationPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

+ func (p *SemanticSegmentationPredictor) ReadPredictedFeatures(ctx context.Context) ([]dlframework.Features, error)

+ func (p SemanticSegmentationPredictor) Modality() (dlframework.Modality, error)

+ func (self *SemanticSegmentationPredictor) Load(ctx context.Context, modelManifest dlframework.ModelManifest, ...) (common.Predictor, error)

+ type Session struct

+ func NewSession(graph *Graph, options *SessionOptions) (*Session, error)

+ func (s *Session) Close() error

+ func (s *Session) ListDevices() ([]Device, error)

+ func (s *Session) Run(ctx context.Context, feeds map[Output]*Tensor, fetches []Output, ...) ([]*Tensor, error)

+ type SessionOptions struct

+ Config []byte

+ Target string

+ type Tensor = tf.Tensor

+ type Trace struct

+ func NewTrace(data *proto.StepStats, graphPath string) (*Trace, error)

+ func (t *Trace) Publish(ctx context.Context, opts ...opentracing.StartSpanOption) error