augmentation

Documentation

Index

• type Cropper
  • func (s *Cropper) GetDataSize() []int
  • func (s *Cropper) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)
• type Scaler
  • func (s *Scaler) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)
• type ShiftMethod
• type Shifter
  • func (s *Shifter) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)

Types

type Cropper

type Cropper struct {
	weight.DataSet
	MaxAmount []int
	// contains filtered or unexported fields
}

func (*Cropper) GetDataSize

func (s *Cropper) GetDataSize() []int

func (*Cropper) GetNextSet

func (s *Cropper) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)

type Scaler

type Scaler struct {
	weight.DataSet
	StDev float64
	Mean  float64
}

func (*Scaler) GetNextSet

func (s *Scaler) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)

type ShiftMethod

type ShiftMethod int

ShiftMethod defines the method used to update parameters in each layer

const (
	//Copy just copies the moved pixels, leaving the old ones where no new pixel is assigned
	Copy ShiftMethod = iota

	//Zeros sets zeros to places where a pixel is not assigned after the shift
	Zeros
)

type Shifter

type Shifter struct {
	weight.DataSet
	MaxAmount []int
	Method    ShiftMethod
	// contains filtered or unexported fields
}

func (*Shifter) GetNextSet

func (s *Shifter) GetNextSet() (*tensor.Tensor, *tensor.Tensor, error)