Book Image

Hands-On Java Deep Learning for Computer Vision

By : Klevis Ramo
Book Image

Hands-On Java Deep Learning for Computer Vision

By: Klevis Ramo

Overview of this book

Although machine learning is an exciting world to explore, you may feel confused by all of its theoretical aspects. As a Java developer, you will be used to telling the computer exactly what to do, instead of being shown how data is generated; this causes many developers to struggle to adapt to machine learning. The goal of this book is to walk you through the process of efficiently training machine learning and deep learning models for Computer Vision using the most up-to-date techniques. The book is designed to familiarize you with neural networks, enabling you to train them efficiently, customize existing state-of-the-art architectures, build real-world Java applications, and get great results in a short space of time. You will build real-world Computer Vision applications, ranging from a simple Java handwritten digit recognition model to real-time Java autonomous car driving systems and face recognition models. By the end of this book, you will have mastered the best practices and modern techniques needed to build advanced Computer Vision Java applications and achieve production-grade accuracy.
Table of Contents (8 chapters)

Building a real-time video, car, and pedestrian detection application

We'll use the YOLO algorithm to build a Java real-time object detection application. We'll use transfer learning to load the YOLO model that was trained on ImageNet and the COCO dataset. Among other objects, it is trying to detect cars, pedestrians, and traffic lights with quite high accuracy.

Architecture of the application

Before jumping into the code, let's see what the architecture of the application will look like:

First, we read the video frames at a certain rate, maybe 30 frames per second. Then, we give each of the frames to the YOLO model, which gives us the bounding-box predictions for each of the objects. Once we have the bounding...