Icon

DL 02 - Train a simple CNN for Image Classification

Train simple CNN

In this workflow we create a simple Convolutional Neural Network using the DL Python Network Creator. We train this network on our image data using the DL Python Network Learner and finally score it using the DL Python Network Executor. The DL Python Network Learner and Executor can be used to write custom training and execution code using Python.

Please note: The workflow series is heavily inspired by the great blog-post of François Chollet (see https://blog.keras.io/building-powerful-image-classification-models-using-very-little-data.html.)

1. Workflow 01 Preprocess image data

2. Workflow 02 Train simple CNN: In this workflow we create a simple Convolutional Neural Network using the DL Python Network Creator. We train this network on our image data using the DL Python Network Learner and finally score it using the DL Python Network Executor. The DL Python Network Learner and Executor can be used to write custom training and execution code using Python.

3. Workflow 03 Fine-tune VGG16 Python
4. Workflow 04 Fine-tune VGG16

In order to run the example, please make sure you have the following KNIME extensions installed:

* KNIME Deep Learning - Keras Integration (Labs)
* KNIME Image Processing (Community Contributions Trusted)
* KNIME Image Processing - Deep Learning Extension (Community Contributions Trusted)
* KNIME Image Processing - Python Extension (Community Contributions Trusted)
* KNIME Streaming Execution (Labs)

You also need a local Python installation that includes Keras. Please refer to https://www.knime.com/deeplearning#keras for installation recommendations and further information.

URL: Keras - Cats vs. Dogs (christian.birkhold) https://hub.knime.com/knime/spaces/Examples/latest/04_Analytics/14_Deep_Learning/02_Keras/04_Cats_and_Dogs~gEkDg2tB3SliwDgo/
URL: Medium: KNIME and Python — Setting up and managing Conda environments https://medium.com/p/2ac217792539

Nodes

Extensions

Links