tensorflowutils

package module

v0.0.0-...-4e22a36 Latest Latest Go to latest Published: Jun 13, 2018 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/combatmage/go-tensorflow-utilities

Links

Open Source Insights

README ¶

go-tensorflow-utilities

Utility functions for image recognition with tensorflow, using golang.

Install

 $ go get github.com/CombatMage/go-tensorflow-utilities

Documentation

Is available on godoc:

https://godoc.org/github.com/CombatMage/go-tensorflow-utilities

Reference

Credit goes to Build an Image Recognition API with Go and TensorFlow.

Documentation ¶

Index ¶

func MakeTensorFromImage(imageBuffer *bytes.Buffer, imageFormat ImageType) (*tf.Tensor, error)
type ImageType
type Label
type Model
- func NewModel(modelFile string, lableFile string) (*Model, error)
- func (model *Model) ClassifyImage(imageBuffer *bytes.Buffer, imageFormat ImageType) ([]Label, error)
- func (model *Model) Run(tensor *tf.Tensor) ([]*tf.Tensor, error)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func MakeTensorFromImage ¶

func MakeTensorFromImage(imageBuffer *bytes.Buffer, imageFormat ImageType) (*tf.Tensor, error)

MakeTensorFromImage converts the given image (as a byte.Buffer) into a tensor. Currently png and jpg is supported as image formats.

Types ¶

type ImageType ¶

type ImageType string

ImageType represents the type of an image, used by MakeTensorFromImage().

const (
	// PNG is a predefined type for MakeTensorFromImage().
	PNG ImageType = "PNG"
	// JPG is a predefined type for MakeTensorFromImage().
	JPG ImageType = "JPG"
)

type Label ¶

type Label struct {
	Label       string
	Probability float32
}

Label represents a classified label with its propability.

type Model ¶

type Model struct {
	SessionModel *tf.Session
	GraphModel   *tf.Graph
	Labels       []string
}

Model represents a loaded tensorflow graph model with it's labels an the generated sessionModel for this graph.

func NewModel ¶

func NewModel(modelFile string, lableFile string) (*Model, error)

NewModel loads graphModel and label from given filepath and returns a new Model, containing the Graph, it's labels and the session. It is assumed that the labels are separated by newlines.

func (*Model) ClassifyImage ¶

func (model *Model) ClassifyImage(imageBuffer *bytes.Buffer, imageFormat ImageType) ([]Label, error)

ClassifyImage tries to classify the given image with the help of this model and returns possible labels with a propability for each label.

func (*Model) Run ¶

func (model *Model) Run(tensor *tf.Tensor) ([]*tf.Tensor, error)

Run evaluates the given tensor with this model and returns the result.

Source Files ¶

View all Source files

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