Skip to content

Commit a9309aa

Browse files
committed
Add repository with fixes for import modules
1 parent 4afade9 commit a9309aa

File tree

11 files changed

+21
-16
lines changed

11 files changed

+21
-16
lines changed

data-pipeline-examples/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ information regarding copyright ownership.
7171
</repository>
7272
<repository>
7373
<id>M2-1</id>
74-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1024</url>
74+
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1028</url>
7575
<releases>
7676
<enabled>true</enabled>
7777
</releases>

dl4j-distributed-training-examples/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ information regarding copyright ownership.
7777
</repository>
7878
<repository>
7979
<id>M2-1</id>
80-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1024</url>
80+
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1028</url>
8181
<releases>
8282
<enabled>true</enabled>
8383
</releases>

dl4j-examples/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ information regarding copyright ownership.
7373
</repository>
7474
<repository>
7575
<id>M2-1</id>
76-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1024</url>
76+
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1028</url>
7777
<releases>
7878
<enabled>true</enabled>
7979
</releases>

nd4j-ndarray-examples/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ information regarding copyright ownership.
5353
</repository>
5454
<repository>
5555
<id>M2-1</id>
56-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1024</url>
56+
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1028</url>
5757
<releases>
5858
<enabled>true</enabled>
5959
</releases>

samediff-examples/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ information regarding copyright ownership.
7676
</repository>
7777
<repository>
7878
<id>M2-1</id>
79-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1024</url>
79+
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1028</url>
8080
<releases>
8181
<enabled>true</enabled>
8282
</releases>

tensorflow-keras-import-examples/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ information regarding copyright ownership.
7070
</repository>
7171
<repository>
7272
<id>M2-1</id>
73-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1024</url>
73+
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1028</url>
7474
<releases>
7575
<enabled>true</enabled>
7676
</releases>

tensorflow-keras-import-examples/src/main/java/org/deeplearning4j/modelimportexamples/keras/advanced/deepmoji/ImportDeepMoji.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@
2222
import org.apache.commons.io.FileUtils;
2323
import org.apache.commons.io.FilenameUtils;
2424
import org.deeplearning4j.nn.graph.ComputationGraph;
25+
import org.deeplearning4j.nn.modelimport.keras.KerasLayer;
26+
import org.deeplearning4j.nn.modelimport.keras.KerasModelImport;
2527
import org.nd4j.linalg.api.ndarray.INDArray;
2628
import org.nd4j.linalg.factory.Nd4j;
2729

tensorflow-keras-import-examples/src/main/java/org/deeplearning4j/modelimportexamples/keras/quickstart/SimpleFunctionalMlpImport.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
import org.nd4j.linalg.api.ndarray.INDArray;
3131
import org.nd4j.linalg.factory.Nd4j;
3232
import org.nd4j.linalg.lossfunctions.LossFunctions;
33+
import org.nd4j.linalg.lossfunctions.impl.LossMCXENT;
3334

3435
import java.io.File;
3536

@@ -108,6 +109,6 @@ public static void main(String[] args) throws Exception {
108109
// layer in this MLP.
109110
Layer loss = model.getLayer(2);
110111
LossLayer lossConf = (LossLayer) loss.conf().getLayer();
111-
assert lossConf.getLossFn().equals(LossFunctions.LossFunction.MCXENT);
112+
assert lossConf.getLossFn() instanceof LossMCXENT;
112113
}
113114
}

tensorflow-keras-import-examples/src/main/java/org/deeplearning4j/modelimportexamples/keras/quickstart/SimpleSequentialMlpImport.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
import org.nd4j.linalg.api.ndarray.INDArray;
3131
import org.nd4j.linalg.factory.Nd4j;
3232
import org.nd4j.linalg.lossfunctions.LossFunctions;
33+
import org.nd4j.linalg.lossfunctions.impl.LossMCXENT;
3334

3435
import java.io.File;
3536

@@ -105,6 +106,6 @@ public static void main(String[] args) throws Exception {
105106
// layer in this MLP.
106107
Layer loss = model.getLayer(2);
107108
LossLayer lossConf = (LossLayer) loss.conf().getLayer();
108-
assert lossConf.getLossFn().equals(LossFunctions.LossFunction.MCXENT);
109+
assert lossConf.getLossFn() instanceof LossMCXENT;
109110
}
110111
}

tensorflow-keras-import-examples/src/main/java/org/deeplearning4j/modelimportexamples/tf/advanced/mobilenet/ImportMobileNetExample.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -29,11 +29,6 @@ public static void main(String[] args) throws Exception {
2929
//new tensorflow import api
3030
TensorflowFrameworkImporter tensorflowFrameworkImporter = new TensorflowFrameworkImporter();
3131

32-
// import the frozen model into a SameDiff instance
33-
//note: this uses the new tensorflow import api
34-
SameDiff sd = tensorflowFrameworkImporter.runImport(modelFile.getAbsolutePath(), Collections.emptyMap());
35-
36-
System.out.println(sd.summary());
3732

3833
System.out.println("\n\n");
3934

@@ -43,6 +38,10 @@ public static void main(String[] args) throws Exception {
4338
// preprocess image with inception preprocessing
4439
INDArray preprocessedImage = inceptionPreprocessing(testImage, 224, 224);
4540

41+
SameDiff sd = tensorflowFrameworkImporter.runImport(modelFile.getAbsolutePath(), Collections.singletonMap("input",preprocessedImage),false);
42+
System.out.println(sd.summary());
43+
44+
4645
// Input and output names are found by looking at sd.summary() (printed earlyer).
4746
// The input variable is the output of no ops, and the output variable is the input of no ops.
4847

@@ -72,7 +71,9 @@ public static void main(String[] args) throws Exception {
7271
static File downloadModel() throws Exception{
7372
String dataDir = FilenameUtils.concat(System.getProperty("user.home"), "dl4j-examples-data/tf_resnet");
7473
String modelFile = FilenameUtils.concat(dataDir, "mobilenet_v2_1.0_224.tgz");
75-
74+
File modelFile2 = new File(modelFile);
75+
if(!modelFile2.exists())
76+
modelFile2.getParentFile().mkdirs();
7677
File frozenFile = new File(FilenameUtils.concat(dataDir, "mobilenet_v2_1.0_224_frozen.pb"));
7778

7879
if(frozenFile.exists()){
@@ -104,7 +105,6 @@ private static INDArray getTestImage() throws IOException {
104105
@SuppressWarnings("SameParameterValue")
105106
private static INDArray inceptionPreprocessing(INDArray img, int height, int width){
106107
// add batch dimension
107-
img = Nd4j.expandDims(img, 0);
108108

109109
// change to channels-last
110110
img = img.permute(0, 2, 3, 1);

0 commit comments

Comments
 (0)