Skip to content

Commit 621020f

Browse files
committed
video sample
Signed-off-by: Robert Altena <[email protected]>
1 parent a96242d commit 621020f

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

dl4j-examples/src/main/java/org/deeplearning4j/examples/recurrent/video/VideoClassificationExample.java

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
/*******************************************************************************
1+
/* *****************************************************************************
22
* Copyright (c) 2015-2019 Skymind, Inc.
33
*
44
* This program and the accompanying materials are made available under the
@@ -23,8 +23,6 @@
2323
import org.datavec.api.split.InputSplit;
2424
import org.datavec.api.split.NumberedFileInputSplit;
2525
import org.deeplearning4j.datasets.datavec.SequenceRecordReaderDataSetIterator;
26-
import org.deeplearning4j.datasets.iterator.AsyncDataSetIterator;
27-
import org.deeplearning4j.eval.Evaluation;
2826
import org.deeplearning4j.nn.conf.BackpropType;
2927
import org.deeplearning4j.nn.conf.GradientNormalization;
3028
import org.deeplearning4j.nn.conf.MultiLayerConfiguration;
@@ -36,8 +34,10 @@
3634
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
3735
import org.deeplearning4j.nn.weights.WeightInit;
3836
import org.deeplearning4j.optimize.listeners.ScoreIterationListener;
37+
import org.nd4j.evaluation.classification.Evaluation;
3938
import org.nd4j.linalg.activations.Activation;
4039
import org.nd4j.linalg.api.ndarray.INDArray;
40+
import org.nd4j.linalg.dataset.AsyncDataSetIterator;
4141
import org.nd4j.linalg.dataset.DataSet;
4242
import org.nd4j.linalg.dataset.api.DataSetPreProcessor;
4343
import org.nd4j.linalg.dataset.api.iterator.DataSetIterator;
@@ -46,6 +46,7 @@
4646
import org.nd4j.linalg.lossfunctions.LossFunctions;
4747

4848
import java.io.File;
49+
import java.nio.charset.Charset;
4950
import java.util.HashMap;
5051
import java.util.Map;
5152

@@ -163,7 +164,7 @@ public static void main(String[] args) throws Exception {
163164
while(trainData.hasNext())
164165
net.fit(trainData.next());
165166
Nd4j.saveBinary(net.params(),new File("videomodel.bin"));
166-
FileUtils.writeStringToFile(new File("videoconf.json"), conf.toJson());
167+
FileUtils.writeStringToFile(new File("videoconf.json"), conf.toJson(), (Charset) null);
167168
System.out.println("Epoch " + i + " complete");
168169

169170
//Evaluate classification performance:
@@ -175,7 +176,7 @@ private static void generateData(String path) throws Exception {
175176
File f = new File(path);
176177
if (!f.exists()) f.mkdir();
177178

178-
/** The data generation code does support the addition of background noise and distractor shapes (shapes which
179+
/* The data generation code does support the addition of background noise and distractor shapes (shapes which
179180
* are shown for one frame only in addition to the target shape) but these are disabled by default.
180181
* These can be enabled to increase the complexity of the learning task.
181182
*/
@@ -223,7 +224,7 @@ private static DataSetIterator getDataSetIterator(String dataDirectory, int star
223224
return new AsyncDataSetIterator(sequenceIter,1);
224225
}
225226

226-
private static SequenceRecordReader getFeaturesReader(String path, int startIdx, int num) throws Exception {
227+
private static SequenceRecordReader getFeaturesReader(String path, int startIdx, int num) {
227228
//InputSplit is used here to define what the file paths look like
228229
InputSplit is = new NumberedFileInputSplit(path + "shapes_%d.mp4", startIdx, startIdx + num - 1);
229230

dl4j-examples/src/main/java/org/deeplearning4j/examples/recurrent/video/VideoGenerator.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
/*******************************************************************************
1+
/* *****************************************************************************
22
* Copyright (c) 2015-2019 Skymind, Inc.
33
*
44
* This program and the accompanying materials are made available under the
@@ -19,6 +19,7 @@
1919
import org.apache.commons.io.FilenameUtils;
2020

2121
import java.awt.*;
22+
import java.nio.charset.StandardCharsets;
2223
import java.nio.file.Files;
2324
import java.nio.file.Paths;
2425
import java.nio.file.StandardOpenOption;
@@ -179,7 +180,7 @@ public static void generateVideoData(String outputFolder, String filePrefix, int
179180
sb.append(labels[j]);
180181
if (j != labels.length - 1) sb.append("\n");
181182
}
182-
Files.write(Paths.get(labelsPath), sb.toString().getBytes("utf-8"), StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
183+
Files.write(Paths.get(labelsPath), sb.toString().getBytes(StandardCharsets.UTF_8), StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
183184
}
184185
}
185186
}

0 commit comments

Comments
 (0)