Skip to content

Commit 2602618

Browse files
committed
More missing repo updates
1 parent 7119e14 commit 2602618

File tree

2 files changed

+8
-25
lines changed

2 files changed

+8
-25
lines changed

dl4j-examples/src/main/java/org/deeplearning4j/examples/advanced/modelling/objectdetection/TinyYoloHouseNumberDetection.java

+5-4
Original file line numberDiff line numberDiff line change
@@ -49,10 +49,12 @@
4949
import org.deeplearning4j.util.ModelSerializer;
5050
import org.deeplearning4j.zoo.model.TinyYOLO;
5151
import org.nd4j.linalg.activations.Activation;
52+
import org.nd4j.linalg.api.memory.enums.DebugMode;
5253
import org.nd4j.linalg.api.ndarray.INDArray;
5354
import org.nd4j.linalg.dataset.api.preprocessor.ImagePreProcessingScaler;
5455
import org.nd4j.linalg.factory.Nd4j;
5556
import org.nd4j.linalg.learning.config.Adam;
57+
import org.nd4j.linalg.profiler.ProfilerConfig;
5658
import org.slf4j.Logger;
5759
import org.slf4j.LoggerFactory;
5860

@@ -122,18 +124,17 @@ public static void main(String[] args) throws java.lang.Exception {
122124
File trainDir = fetcher.getDataSetPath(DataSetType.TRAIN);
123125
File testDir = fetcher.getDataSetPath(DataSetType.TEST);
124126

125-
126127
log.info("Load data...");
127128

128129
FileSplit trainData = new FileSplit(trainDir, NativeImageLoader.ALLOWED_FORMATS, rng);
129130
FileSplit testData = new FileSplit(testDir, NativeImageLoader.ALLOWED_FORMATS, rng);
130131

131132
ObjectDetectionRecordReader recordReaderTrain = new ObjectDetectionRecordReader(height, width, nChannels,
132-
gridHeight, gridWidth, new SvhnLabelProvider(trainDir));
133+
gridHeight, gridWidth, new SvhnLabelProvider(trainDir));
133134
recordReaderTrain.initialize(trainData);
134135

135136
ObjectDetectionRecordReader recordReaderTest = new ObjectDetectionRecordReader(height, width, nChannels,
136-
gridHeight, gridWidth, new SvhnLabelProvider(testDir));
137+
gridHeight, gridWidth, new SvhnLabelProvider(testDir));
137138
recordReaderTest.initialize(testData);
138139

139140
// ObjectDetectionRecordReader performs regression, so we need to specify it here
@@ -210,7 +211,7 @@ public static void main(String[] args) throws java.lang.Exception {
210211
CanvasFrame frame = new CanvasFrame("HouseNumberDetection");
211212
OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
212213
org.deeplearning4j.nn.layers.objdetect.Yolo2OutputLayer yout =
213-
(org.deeplearning4j.nn.layers.objdetect.Yolo2OutputLayer)model.getOutputLayer(0);
214+
(org.deeplearning4j.nn.layers.objdetect.Yolo2OutputLayer)model.getOutputLayer(0);
214215
List<String> labels = train.getLabels();
215216
test.setCollectMetaData(true);
216217
Scalar[] colormap = {RED,BLUE,GREEN,CYAN,YELLOW,MAGENTA,ORANGE,PINK,LIGHTBLUE,VIOLET};

tensorflow-keras-import-examples/pom.xml

+3-21
Original file line numberDiff line numberDiff line change
@@ -61,27 +61,9 @@ information regarding copyright ownership.
6161
<updatePolicy>daily</updatePolicy> <!-- Optional, update daily -->
6262
</snapshots>
6363
</repository>
64-
<repository>
65-
<id>M2-2</id>
66-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1040</url>
67-
<releases>
68-
<enabled>true</enabled>
69-
</releases>
70-
</repository>
71-
<repository>
72-
<id>M2-3</id>
73-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1049</url>
74-
<releases>
75-
<enabled>true</enabled>
76-
</releases>
77-
</repository>
78-
<repository>
79-
<id>M2-4</id>
80-
<url>https://s01.oss.sonatype.org/content/repositories/orgdeeplearning4j-1061</url>
81-
<releases>
82-
<enabled>true</enabled>
83-
</releases>
84-
</repository>
64+
65+
66+
8567
</repositories>
8668

8769

0 commit comments

Comments
 (0)