Java Deep Learning Projects

Step 1 – Loading a pre-trained YOLO model

Since Alpha release 1.0.0, DL4J provides a Tiny YOLO model via ZOO. For this, we need to add a dependency to your Maven friendly pom.xml file:

<dependency>  <groupId>org.deeplearning4j</groupId>  <artifactId>deeplearning4j-zoo</artifactId>  <version>${dl4j.version}</version></dependency>

Apart from this, if possible, make sure you utilize the CUDA and cuDNN by adding the following dependencies (see Chapter 2, Cancer Types Prediction Using Recurrent Type Networks, for more details):

<dependency>  <groupId>org.nd4j</groupId>  <artifactId>nd4j-cuda-9.0-platform</artifactId>  <version>${nd4j.version}</version></dependency><dependency>  <groupId>org.deeplearning4j</groupId> <artifactId>deeplearning4j-cuda-9.0</artifactId> ...