DL4J: How to create a neural network that draws images – Step by step guide

In this guide we will create a neural network, which we will put the task of transforming one image into another. The application will be built on the basis of the DL4J application template described in the previous article based on JavaFx and Spring Boot. Thanks to the fact that we will use it, the amount of code necessary to implement has been reduced to only one controller and one form.

Despite the small amount of code, we will be able to get through the most important issues of machine learning. We will start the journey by preparing training and test data. In the following, we will create a neural network model fitted to our needs, which we will then learn to solve the task set for it. Finally, we ask the network for the result that we will visualize as a picture.

The applied template of the application allows us to track the progress of learning, and to visualize the current result of the network during the learning process. Below is a screenshot of the application after doing nearly 100,000 iterations of the learning process.