mtcnn

Documentation

Index

• func CropResizeImage(img *tf.Tensor, bbox [][]float32, size []int32) (*tf.Tensor, error)
• func TensorFromJpeg(bytes []byte) (*tf.Tensor, error)
• type MtcnnDetector
  • func NewMtcnnDetector(modelFile string) (*MtcnnDetector, error)
  • func (det *MtcnnDetector) Close()
  • func (det *MtcnnDetector) Config(scaleFactor, minSize float64, scoreThresholds []float32)
  • func (det *MtcnnDetector) DetectFaces(tensor *tf.Tensor) ([][]float32, error)

Functions

func CropResizeImage

func CropResizeImage(img *tf.Tensor, bbox [][]float32, size []int32) (*tf.Tensor, error)

func TensorFromJpeg

func TensorFromJpeg(bytes []byte) (*tf.Tensor, error)

Types

type MtcnnDetector

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

func NewMtcnnDetector

func NewMtcnnDetector(modelFile string) (*MtcnnDetector, error)

func (*MtcnnDetector) Close

func (det *MtcnnDetector) Close()

func (*MtcnnDetector) Config

func (det *MtcnnDetector) Config(scaleFactor, minSize float64, scoreThresholds []float32)

func (*MtcnnDetector) DetectFaces

func (det *MtcnnDetector) DetectFaces(tensor *tf.Tensor) ([][]float32, error)