Skip to content

Instantly share code, notes, and snippets.

@tomthetrainer
Created March 27, 2018 18:42
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tomthetrainer/c6d994cfeca34378598c3d699f38af0b to your computer and use it in GitHub Desktop.
Save tomthetrainer/c6d994cfeca34378598c3d699f38af0b to your computer and use it in GitHub Desktop.
inception import
package ai.skymind.training.solutions;
import org.datavec.api.util.ClassPathResource;
import org.datavec.image.loader.NativeImageLoader;
import org.deeplearning4j.nn.conf.ComputationGraphConfiguration;
import org.deeplearning4j.nn.graph.ComputationGraph;
import org.deeplearning4j.nn.modelimport.keras.KerasModelImport;
import org.deeplearning4j.nn.modelimport.keras.trainedmodels.TrainedModels;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.dataset.api.preprocessor.DataNormalization;
import org.nd4j.linalg.dataset.api.preprocessor.VGG16ImagePreProcessor;
import org.nd4j.linalg.factory.Nd4j;
import java.io.File;
/**
* Created by tomhanlon on 4/20/17.
*/
public class KerasModelimportInceptionV3 {
public static void main(String[] args) throws Exception{
//Path to Saved Model and weights
// We use DataVec's ClassPathResource here, you could pass the import files path string directly
String kerasModelfromKerasExport = new ClassPathResource("inception_v3_complete.h5").getFile().getPath();
/*
Create a MultiLayerNetwork from the saved model
*/
// MultiLayerNetwork model = org.deeplearning4j.nn.modelimport.keras.KerasModelImport.importKerasSequentialModelAndWeights(kerasModelfromKerasExport);
// ComputationGraph model = KerasModelImport.importKerasModelAndWeights(kerasModelfromKerasExport);
//ComputationGraph model = KerasModelImport.importKerasModelAndWeights(kerasModelfromKerasExport);
ComputationGraph model = KerasModelImport.importKerasModelAndWeights("/Users/tomhanlon/tensorflow/vgg16/keras-model-zoo/deep-learning-models/inception_V3_config","/Users/tomhanlon/tensorflow/vgg16/keras-model-zoo/deep-learning-models/inception_v3.h5",false);
//ComputationGraph model = new ComputationGraph()
// File elephant = new ClassPathResource("elephant.jpeg").getFile();
//File file = new File("/Users/tomhanlon/tensorflow/vgg16/keras-model-zoo/deep-learning-models/images/lion.jpeg");
//NativeImageLoader loader = new NativeImageLoader(299, 299, 3);
//NativeImageLoader loader = new NativeImageLoader(10, 10, 3);
//INDArray image = loader.asMatrix(elephant);
//System.out.print(image);
//DataNormalization scaler = new
//scaler.transform(image);
//DataNormalization scaler = new VGG16ImagePreProcessor();
//scaler.transform(image);
//System.out.print(image);
//INDArray[] output = model.output(false,image);
//System.out.println(TrainedModels.VGG16.decodePredictions(output[0]));
//System.out.println(output);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment