maskrcnn

Documentation

Index

• type Detections
• type Masks
• type RCNN
  • func NewRCNN(modelPath, tagName string) (*RCNN, error)
  • func (r *RCNN) Close() error
  • func (r *RCNN) Detect(img image.Image) ([]gildasai.Mask, error)

Types

type Detections

type Detections struct {
	Values [][][]float32
}

type Masks

type Masks struct {
	Values [][][][][]float32
}

type RCNN

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

func NewRCNN

func NewRCNN(modelPath, tagName string) (*RCNN, error)

func (*RCNN) Close

func (r *RCNN) Close() error

func (*RCNN) Detect

func (r *RCNN) Detect(img image.Image) ([]gildasai.Mask, error)