tensorflowAPI

package

v0.0.0-...-d54de9c Latest Latest Go to latest Published: Feb 15, 2023 License: MIT Imports: 12 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/l-pavlova/image-master

Links

Open Source Insights

Documentation ¶

Index ¶

Constants
type Label
type Model
type TensorFlowClient
- func NewTensorFlowClient(logger logging.ImageMasterLogger) *TensorFlowClient
- func (t *TensorFlowClient) ClassifyImage(img image.Image) ([]Label, []float32, [][]float32, error)

Constants ¶

View Source

const (
	INCEPTION_MODEL_PATH          = "tensorflowAPI/model/tensorflow_inception_graph.pb"
	INCEPTION_MODEL_LABELS        = "tensorflowAPI/model/imagenet_comp_graph_label_strings.txt"
	COCO_MODEL_PATH               = "tensorflowAPI/model/frozen_inference_graph.pb"
	COCO_LABELS_PATH              = "tensorflowAPI/model/coco_labels.txt"
	DEPTH_OF_LABEL_CLASSIFICATION = 5
)

Variables ¶

This section is empty.

Functions ¶

This section is empty.

Types ¶

type Label ¶

type Label struct {
	Label       string
	Probability float32
}

type Model ¶

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

type TensorFlowClient ¶

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

func NewTensorFlowClient ¶

func NewTensorFlowClient(logger logging.ImageMasterLogger) *TensorFlowClient

func (*TensorFlowClient) ClassifyImage ¶

func (t *TensorFlowClient) ClassifyImage(img image.Image) ([]Label, []float32, [][]float32, error)

function that takes an image and returns the top probable objects in it

Source Files ¶

View all Source files

tensorflowAPI.go

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