models

Documentation

Index

• type Coco
  • func NewCoco() *Coco
  • func (c *Coco) CloseSession()
  • func (c *Coco) Load() error
  • func (c *Coco) Predict(data []byte) *responses.ObjectDetectionResponse
• type Rcnn
  • func NewRcnn() *Rcnn
  • func (c *Rcnn) CloseSession()
  • func (c *Rcnn) Load() error
  • func (c *Rcnn) Predict(data []byte) *responses.ObjectDetectionResponse

Types

type Coco

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

Coco contains tensorflow model and human-readable string labels

func NewCoco

func NewCoco() *Coco

NewCoco returns a Coco object

func (*Coco) CloseSession

func (c *Coco) CloseSession()

CloseSession closes a session

func (*Coco) Load

func (c *Coco) Load() error

Load loads the ssd_mobilenet_v1_coco_2018_01_28 SavedModel.

func (*Coco) Predict

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

Predict predicts labels for an input image

type Rcnn

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

Rcnn contains tensorflow model and human-readable string labels

func NewRcnn

func NewRcnn() *Rcnn

NewRcnn returns a Rcnn object

func (*Rcnn) CloseSession

func (c *Rcnn) CloseSession()

CloseSession closes a session

func (*Rcnn) Load

func (c *Rcnn) Load() error

Load loads the ssd_mobilenet_v1_Rcnn_2018_01_28 SavedModel.

func (*Rcnn) Predict

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

Predict predicts labels for an input image