predictor

package
v0.0.0-...-8f82a34 Latest Latest
Warning

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

Go to latest
Published: Jun 29, 2020 License: NCSA Imports: 26 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func NewImageClassificationPredictor

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

New ...

func NewImageEnhancementPredictor

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

New ...

func NewObjectDetectionPredictor

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

New ...

func NewSemanticSegmentationPredictor

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

Types

type ImageClassificationPredictor

type ImageClassificationPredictor struct {
	common.ImagePredictor
	// contains filtered or unexported fields
}

ImageClassificationPredictor ...

func (*ImageClassificationPredictor) Close

Close ...

func (*ImageClassificationPredictor) Download

Download ...

func (*ImageClassificationPredictor) Load

Load ...

func (*ImageClassificationPredictor) Modality

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 ...

func (*ImageClassificationPredictor) Reset

Reset ...

type ImageEnhancementPredictor

type ImageEnhancementPredictor struct {
	common.ImagePredictor
	// contains filtered or unexported fields
}

ImageEnhancementPredictor ...

func (*ImageEnhancementPredictor) Close

func (p *ImageEnhancementPredictor) Close() error

func (*ImageEnhancementPredictor) Download

Download ...

func (ImageEnhancementPredictor) GetInputLayerName

func (p ImageEnhancementPredictor) GetInputLayerName(reader io.Reader, layer string) (string, error)

func (ImageEnhancementPredictor) GetOutputLayerName

func (p ImageEnhancementPredictor) GetOutputLayerName(reader io.Reader, layer string) (string, error)

func (*ImageEnhancementPredictor) Load

Load ...

func (ImageEnhancementPredictor) Modality

func (*ImageEnhancementPredictor) Predict

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

Predict ...

func (*ImageEnhancementPredictor) ReadPredictedFeatures

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

ReadPredictedFeatures ...

func (*ImageEnhancementPredictor) Reset

type ObjectDetectionPredictor

type ObjectDetectionPredictor struct {
	common.ImagePredictor
	// contains filtered or unexported fields
}

ObjectDetectionPredictor ...

func (*ObjectDetectionPredictor) Close

func (p *ObjectDetectionPredictor) Close() error

Close ...

func (*ObjectDetectionPredictor) Download

Download ...

func (*ObjectDetectionPredictor) GetInputLayerName

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

func (*ObjectDetectionPredictor) GetOutputLayerName

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

func (*ObjectDetectionPredictor) Load

Load ...

func (*ObjectDetectionPredictor) Modality

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 ...

func (*ObjectDetectionPredictor) Reset

Reset ...

type SemanticSegmentationPredictor

type SemanticSegmentationPredictor struct {
	common.ImagePredictor
	// contains filtered or unexported fields
}

func (*SemanticSegmentationPredictor) Close

func (*SemanticSegmentationPredictor) Download

Download ...

func (*SemanticSegmentationPredictor) GetInputLayerName

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

func (*SemanticSegmentationPredictor) GetOutputLayerName

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

func (*SemanticSegmentationPredictor) Load

func (SemanticSegmentationPredictor) Modality

func (*SemanticSegmentationPredictor) Predict

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

Predict ...

func (*SemanticSegmentationPredictor) ReadPredictedFeatures

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

ReadPredictedFeatures ...

func (*SemanticSegmentationPredictor) Reset

Source Files

  • image_classification_predictor.go
  • image_enhancement_predictor.go
  • image_instance_segmentation_predictor.go
  • image_object_detection_predictor.go
  • image_semantic_segmentation_predictor.go
  • init.go
  • utils.go

Jump to

Keyboard shortcuts

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