vggface2

Documentation

Index

• func VGGFace2Builder(opts Opts) func(*godl.Model) godl.Module
• type BlockOpts
• type ConvBlockModule
  • func ConvBlock(m *godl.Model, opts BlockOpts) *ConvBlockModule
  • func (m *ConvBlockModule) Forward(inputs ...*godl.Node) godl.Nodes
  • func (m *ConvBlockModule) Name() string
• type IdentityBlockModule
  • func IdentityBlock(m *godl.Model, opts BlockOpts) *IdentityBlockModule
  • func (m *IdentityBlockModule) Forward(inputs ...*godl.Node) godl.Nodes
  • func (m *IdentityBlockModule) Name() string
• type Opts
• type VGGFace2Module
  • func VGGFace2(m *godl.Model, opts Opts) *VGGFace2Module
  • func (m *VGGFace2Module) Forward(inputs ...*godl.Node) godl.Nodes

Functions

func VGGFace2Builder

func VGGFace2Builder(opts Opts) func(*godl.Model) godl.Module

Types

type BlockOpts

type BlockOpts struct {
	Filters    [3]int
	Stride     []int
	Stage      int
	KernelSize tensor.Shape
	Block      int
}

type ConvBlockModule

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

func ConvBlock

func ConvBlock(m *godl.Model, opts BlockOpts) *ConvBlockModule

func (*ConvBlockModule) Forward

func (m *ConvBlockModule) Forward(inputs ...*godl.Node) godl.Nodes

func (*ConvBlockModule) Name

func (m *ConvBlockModule) Name() string

type IdentityBlockModule

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

func IdentityBlock

func IdentityBlock(m *godl.Model, opts BlockOpts) *IdentityBlockModule

func (*IdentityBlockModule) Forward

func (m *IdentityBlockModule) Forward(inputs ...*godl.Node) godl.Nodes

func (*IdentityBlockModule) Name

func (m *IdentityBlockModule) Name() string

type Opts

type Opts struct {
	WithBias              bool
	WeightsInit, BiasInit gorgonia.InitWFn
	Learnable             bool

	PreTrained            bool
	OnlyFeatureExtraction bool
	Classes               int
}

type VGGFace2Module

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

func VGGFace2

func VGGFace2(m *godl.Model, opts Opts) *VGGFace2Module

func (*VGGFace2Module) Forward

func (m *VGGFace2Module) Forward(inputs ...*godl.Node) godl.Nodes