styletransfer

Documentation

Overview

Package styletransfer include style transfer

Index

• func Transform(d StyleTransfer, img *common.Image, out *common.Image) error
• type AnimeGan2
  • func NewAnimeGan2() *AnimeGan2
  • func (d *AnimeGan2) Destroy()
  • func (d *AnimeGan2) LoadModel(modelPath string) error
  • func (d *AnimeGan2) Pointer() unsafe.Pointer
  • func (d *AnimeGan2) Transform(img *common.Image, out *common.Image) error
• type StyleTransfer

Functions

func Transform

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

Transform returns style transform image

Types

type AnimeGan2

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

AnimeGan2 represents animegan2 style transfer

func NewAnimeGan2

func NewAnimeGan2() *AnimeGan2

NewAnimeGan2 returns a new AnimeGan2

func (*AnimeGan2) Destroy

func (d *AnimeGan2) Destroy()

Destroy free style transfer

func (*AnimeGan2) LoadModel

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

LoadModel load model for detecter

func (*AnimeGan2) Pointer

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

Pointer implement Estimator interface

func (*AnimeGan2) Transform

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

Transform implement StyleTransfer interface

type StyleTransfer

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

StyleTransfer represents Style Transfer interface