tfmodel

package

v0.0.0-...-9b9b30a Latest Latest Go to latest Published: Aug 28, 2021 License: MIT Imports: 9 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/pastelnetwork/gonode

Documentation ¶

Index ¶

Variables
type Config
- func NewConfig(path, input string) Config
type List
- func NewList(configs []Config) List
- func (models *List) Exec(ctx context.Context, value interface{}) ([][]float32, error)
- func (models *List) Load(ctx context.Context, baseDir string) error
type TFModel
- func NewModel(info Config) *TFModel

Constants ¶

This section is empty.

Variables ¶

View Source

var (
	EfficientNetB7    = NewConfig("EfficientNetB7.tf", "serving_default_input_1")
	EfficientNetB6    = NewConfig("EfficientNetB6.tf", "serving_default_input_2")
	InceptionResNetV2 = NewConfig("InceptionResNetV2.tf", "serving_default_input_3")
	DenseNet201       = NewConfig("DenseNet201.tf", "serving_default_input_4")
	InceptionV3       = NewConfig("InceptionV3.tf", "serving_default_input_5")
	NASNetLarge       = NewConfig("NASNetLarge.tf", "serving_default_input_6")
	ResNet152V2       = NewConfig("ResNet152V2.tf", "serving_default_input_7")
)

Configs of the available models.

View Source

var AllConfigs = []Config{
	EfficientNetB7,
	EfficientNetB6,
	InceptionResNetV2,
	DenseNet201,
	InceptionV3,
	NASNetLarge,
	ResNet152V2,
}

AllConfigs contains configs of all models.

Functions ¶

This section is empty.

Types ¶

type Config ¶

type Config struct {
	// contains filtered or unexported fields
}

Config represents information about the exported model.

func NewConfig ¶

func NewConfig(path, input string) Config

NewConfig returns a new Config instantce.

type List ¶

type List []*TFModel

List represents multiple TFModel.

func NewList ¶

func NewList(configs []Config) List

NewList returns a new List instance.

func (*List) Exec ¶

func (models *List) Exec(ctx context.Context, value interface{}) ([][]float32, error)

Exec executes the nodes/tensors that must be present in the loaded models.

func (*List) Load ¶

func (models *List) Load(ctx context.Context, baseDir string) error

Load loads models data from the baseDir.

type TFModel ¶

type TFModel struct {
	sync.Mutex

	Config
	// contains filtered or unexported fields
}

TFModel represents tensorflow exported model.

func NewModel ¶

func NewModel(info Config) *TFModel

NewModel returns a new TFModel instance.

func (*TFModel) Exec ¶

func (model *TFModel) Exec(ctx context.Context, value interface{}) ([]float32, error)

Exec executes the nodes/tensors that must be present in the loaded model.

func (*TFModel) Load ¶

func (model *TFModel) Load(ctx context.Context, baseDir string) error

Load loads model data from the baseDir. The graph loaded is identified by the set of tags specified when exporting it.

func (*TFModel) String ¶

func (model *TFModel) String() string

String implements fmt.Stringer.String

Source Files ¶

View all Source files

Directories ¶

Path	Synopsis
test

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