DEV Community

Cover image for Creating your custom model for Tensorflow.JS
Daniel Persson
Daniel Persson

Posted on

Creating your custom model for Tensorflow.JS

We look into how to use Java to create an image classification model using TFRecords and train it with python. Then we will read that model in TensorflowJS in order to predict images.

Repository:
https://github.com/kalaspuffar/dataset-tensorflow-js

Discussion (0)