styletransfer

package

v1.0.0 Latest Latest Go to latest Published: Nov 11, 2021 License: Apache-2.0 Imports: 5 Imported by: 0

Details

Package styletransfer include style transfer

This section is empty.

This section is empty.

func Transform(d StyleTransfer, img *common.Image) (image.Image, error)

Transform returns style transform image

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

AnimeGan2 represents animegan2 style transfer

func NewAnimeGan2() *AnimeGan2

NewAnimeGan2 returns a new AnimeGan2

func (d *AnimeGan2) Destroy()

Destroy free style transfer

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

LoadModel load model for detecter

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

Pointer implement Estimator interface

func (d *AnimeGan2) Transform(img *common.Image) (image.Image, error)

Transform implement StyleTransfer interface

type StyleTransfer interface {
	common.Estimator
	Transform(img *common.Image) (image.Image, error)
}

StyleTransfer represents Style Transfer interface