编码
- 在父工程 dl4j-tutorials下新建名为 simple-convolution的子工程,其pom.xml如下,可见这里的dl4j版本被指定为1.0.0-beta6:
<project xmlns=“http://maven.apache.org/POM/4.0.0”
xmlns:xsi=“http://www.w3.org/2001/XMLSchema-instance”
xsi:schemaLocation=“http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd”>
dlfj-tutorials
com.bolingcavalry
1.0-SNAPSHOT
4.0.0
simple-convolution
<dl4j-master.version>1.0.0-beta6</dl4j-master.version>
org.projectlombok
lombok
ch.qos.logback
logback-classic
org.deeplearning4j
deeplearning4j-core
${dl4j-master.version}
org.nd4j
${nd4j.backend}
${dl4j-master.version}
- 接下来按照前面的分析实现代码,已经添加了详细注释,就不再赘述了:
package com.bolingcavalry.convolution;
import lombok.extern.slf4j.Slf4j;
import org.datavec.api.io.labels.ParentPathLabelGenerator;
import org.datavec.api.split.FileSplit;
import org.datavec.image.loader.NativeImageLoader;
import org.datavec.image.recordreader.ImageRecordReader;
import org.deeplearning4j.datasets.datavec.RecordReaderDataSetIterator;
import org.dee