models

package

v0.0.0-...-8cdc3bd Latest Latest Go to latest Published: Oct 19, 2021 License: MIT Imports: 9 Imported by: 1

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

This section is empty.

This section is empty.

This section is empty.

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

Coco contains tensorflow model and human-readable string labels

func NewCoco() *Coco

NewCoco returns a Coco object

func (c *Coco) CloseSession()

CloseSession closes a session

func (c *Coco) Load() error

Load loads the ssd_mobilenet_v1_coco_2018_01_28 SavedModel.

func (c *Coco) Predict(data []byte) *responses.ObjectDetectionResponse

Predict predicts labels for an input image

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

Rcnn contains tensorflow model and human-readable string labels

func NewRcnn() *Rcnn

NewRcnn returns a Rcnn object

func (c *Rcnn) CloseSession()

CloseSession closes a session

func (c *Rcnn) Load() error

Load loads the ssd_mobilenet_v1_Rcnn_2018_01_28 SavedModel.

func (c *Rcnn) Predict(data []byte) *responses.ObjectDetectionResponse

Predict predicts labels for an input image