mcjojos / inception-serving-sb

Tensorflow serving in Spring Boot

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Zero to image recognition in 60 seconds with Tensorflow and Spring Boot

Build Status Coverage Status

TL;DR: for the impatient ones wanting to have a web service for image recognition without any Tensorflow prerequisites: run ./gradlew fetchInceptionFrozenModel bootrun, navigate to http://localhost:8080 and upload an image. The backend will categorize the image and output the result along with the probability.

Screenshot (non clickable)

Why

Tensorflow is hard enough to wrap one's head around. It has several parts that deal with preparing data, defining and training a model and finally, outputting a model that can then be used to categorize (infer) other data. There's math involved, new vocabulary to learn and on top, a toolchain which revolves around Python. This project only addresses serving a Tensorflow pre-trained image categorization model, otherwise called the Inception model.

Prerequisites

  • JDK 8

Run

./gradlew fetchInceptionFrozenModel bootrun

Navigate to http://localhost:8080 and upload an image. The backend will categorize the image and output the result along with the probability.

How

Head to the blog post for the ful monty.

About

Tensorflow serving in Spring Boot

License:MIT License


Languages

Language:Java 73.3%Language:HTML 19.6%Language:JavaScript 7.1%