Examples of CIFAR10TrainingInputProvider


Examples of com.github.neuralnetworks.samples.cifar.CIFARInputProvider.CIFAR10TrainingInputProvider

    public void testSigmoidBP() {
  Environment.getInstance().setUseDataSharedMemory(false);
  Environment.getInstance().setUseWeightsSharedMemory(false);
  NeuralNetworkImpl mlp = NNFactory.mlpSigmoid(new int[] { 3072, 10 }, true);

  CIFAR10TrainingInputProvider trainInputProvider = new CIFAR10TrainingInputProvider("cifar-10-batches-bin"); // specify your own path
  trainInputProvider.getProperties().setGroupByChannel(true);
  trainInputProvider.getProperties().setScaleColors(true);
  trainInputProvider.addInputModifier(new ScalingInputFunction(255));

  // specify your own path
  CIFAR10TestingInputProvider testInputProvider = new CIFAR10TestingInputProvider("cifar-10-batches-bin"); // specify your own path
  testInputProvider.getProperties().setGroupByChannel(true);
  testInputProvider.getProperties().setScaleColors(true);
View Full Code Here
TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.