Search in sources :

Example 1 with ImgMinSizeLayer

use of com.simiacryptus.mindseye.layers.cudnn.ImgMinSizeLayer in project MindsEye by SimiaCryptus.

the class VGG16_HDF5 method phase0.

/**
 * Phase 0.
 */
protected void phase0() {
    add(new ImgMinSizeLayer(226, 226));
    add(new ImgBandBiasLayer(3).set(new Tensor(-103.939, -116.779, -123.68)));
}
Also used : ImgBandBiasLayer(com.simiacryptus.mindseye.layers.cudnn.ImgBandBiasLayer) Tensor(com.simiacryptus.mindseye.lang.Tensor) ImgMinSizeLayer(com.simiacryptus.mindseye.layers.cudnn.ImgMinSizeLayer)

Example 2 with ImgMinSizeLayer

use of com.simiacryptus.mindseye.layers.cudnn.ImgMinSizeLayer in project MindsEye by SimiaCryptus.

the class VGG19_HDF5 method phase0.

/**
 * Phase 0.
 */
protected void phase0() {
    add(new ImgMinSizeLayer(226, 226));
    add(new ImgBandBiasLayer(3).set(new Tensor(-103.939, -116.779, -123.68)));
}
Also used : ImgBandBiasLayer(com.simiacryptus.mindseye.layers.cudnn.ImgBandBiasLayer) Tensor(com.simiacryptus.mindseye.lang.Tensor) ImgMinSizeLayer(com.simiacryptus.mindseye.layers.cudnn.ImgMinSizeLayer)

Aggregations

Tensor (com.simiacryptus.mindseye.lang.Tensor)2 ImgBandBiasLayer (com.simiacryptus.mindseye.layers.cudnn.ImgBandBiasLayer)2 ImgMinSizeLayer (com.simiacryptus.mindseye.layers.cudnn.ImgMinSizeLayer)2