segmentor

package
v1.0.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Feb 7, 2022 License: Apache-2.0 Imports: 4 Imported by: 0

Documentation

Overview

Package segmentor pose estimator

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Matting

func Matting(d Segmentor, img *common.Image, out *common.Image) error

Matting returns pose segment matting image

func Merge

func Merge(d Segmentor, img *common.Image, bg *common.Image, out *common.Image) error

Merge merge pose with background

Types

type Deeplabv3plus

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

Deeplabv3plus represents deeplabv3plus segmentor

func NewDeeplabv3plus

func NewDeeplabv3plus() *Deeplabv3plus

NewDeeplabv3plus returns a new Deeplabv3plus

func (*Deeplabv3plus) Destroy

func (d *Deeplabv3plus) Destroy()

Destroy free segmentor

func (*Deeplabv3plus) LoadModel

func (d *Deeplabv3plus) LoadModel(modelPath string) error

LoadModel load model for detecter

func (*Deeplabv3plus) Matting

func (d *Deeplabv3plus) Matting(img *common.Image, out *common.Image) error

Matting implement Segmentor interface

func (*Deeplabv3plus) Merge

func (d *Deeplabv3plus) Merge(img *common.Image, bg *common.Image, out *common.Image) error

Merge implement Segmentor interface

func (*Deeplabv3plus) Pointer

func (d *Deeplabv3plus) Pointer() unsafe.Pointer

Pointer implement Estimator interface

type ERDNet

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

ERDNet represents erdnet segmentor

func NewERDNet

func NewERDNet() *ERDNet

NewERDNet returns a new ERDNet

func (*ERDNet) Destroy

func (d *ERDNet) Destroy()

Destroy free segmentor

func (*ERDNet) LoadModel

func (d *ERDNet) LoadModel(modelPath string) error

LoadModel load model for detecter

func (*ERDNet) Matting

func (d *ERDNet) Matting(img *common.Image, out *common.Image) error

Matting implement Segmentor interface

func (*ERDNet) Merge

func (d *ERDNet) Merge(img *common.Image, bg *common.Image, out *common.Image) error

Merge implement Segmentor interface

func (*ERDNet) Pointer

func (d *ERDNet) Pointer() unsafe.Pointer

Pointer implement Estimator interface

type RVM

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

RVM represents rvm segmentor

func NewRVM

func NewRVM(width int, height int) *RVM

NewRVM returns a new RVM

func (*RVM) Destroy

func (d *RVM) Destroy()

Destroy free segmentor

func (*RVM) LoadModel

func (d *RVM) LoadModel(modelPath string) error

LoadModel load model for detecter

func (*RVM) Matting

func (d *RVM) Matting(img *common.Image, out *common.Image) error

Matting implement Segmentor interface

func (*RVM) Merge

func (d *RVM) Merge(img *common.Image, bg *common.Image, out *common.Image) error

Merge implement Segmentor interface

func (*RVM) Pointer

func (d *RVM) Pointer() unsafe.Pointer

Pointer implement Estimator interface

type Segmentor

type Segmentor interface {
	common.Estimator
	Matting(img *common.Image, out *common.Image) error
	Merge(img *common.Image, bg *common.Image, out *common.Image) error
}

Segmentor represents segmentor interface

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL