predict

Jan 10, 2024

Changes in this version

+ var ImageHeight int32 = 128

+ var ImageWidth int32 = 128

+ type Predictor struct

+ Model *tf.SavedModel

+ func NewPredictor(modeldir string) (Predictor, error)

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

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

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