O'Reilly logo

Java Deep Learning Projects by Md. Rezaul Karim

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

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 2Cancer 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> ...

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required