predict

package

v1.0.0 Latest Latest Go to latest Published: Dec 4, 2023 License: MIT Imports: 6 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

View Source

const ImageHeight = 128

View Source

const ImageWidth = 128

This section is empty.

This section is empty.

type Predictor struct {
	Model *tf.SavedModel
}

func NewPredictor(modeldir string) (Predictor, error)

declare method

func (p Predictor) Predict(imagefile string, model *tf.SavedModel) (int, error)

Predict - Return index with highest probability

func (p Predictor) PredictWithDeviation(imagefile string, model *tf.SavedModel) (classpred int, deviation float32, err error)

Predict - return class prediction and standard deviation

func (p Predictor) PredictionsArr(imagefile string, model *tf.SavedModel) ([]float32, error)

PredictionsArr Return array of probabilities