Options
All
  • Public
  • Public/Protected
  • All
Menu

Class DataAugmentationLayer

Hierarchy

  • Layer
    • DataAugmentationLayer

Index

Constructors

constructor

Properties

Private augmentation_kernel

augmentation_kernel: any

data_mean

data_mean: Matrix

delta_params

delta_params: string[]

input_klass

input_klass: string

need_update

need_update: boolean

out_shape

out_shape: number[]

random_crop

random_crop: boolean

random_flip

random_flip: boolean

scale

scale: number

train_params

train_params: string[]

Methods

Private _get_data_mean

  • _get_data_mean(devicetype: string): Matrix

backward

  • backward(bottoms: Matrix[], top_deltas: Matrix[], config: ForwardConfiguration, callback: function): void

calculateUpdateParams

  • calculateUpdateParams(bottoms: Matrix[], top_deltas: Matrix[], config: ForwardConfiguration, callback: function): void

destruct

  • destruct(): void

forward

Private get_kernel

  • get_kernel(): any

init

  • init(callback: function): void

release

  • release(): void

set_data_mean

  • set_data_mean(data_mean: Matrix): void

to_cl

  • to_cl(): void

to_cpu

  • to_cpu(): void

Generated using TypeDoc