DEV Community

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

Posted on

1

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

Top comments (0)

Billboard image

Create up to 10 Postgres Databases on Neon's free plan.

If you're starting a new project, Neon has got your databases covered. No credit cards. No trials. No getting in your way.

Try Neon for Free →

Instrument, monitor, fix: a hands-on debugging session

Join Lazar for a hands-on session where you’ll build it, break it, debug it, and fix it. You’ll set up Sentry, track errors, use Session Replay and Tracing, and leverage some good ol’ AI to find and fix issues fast.

Tune in to the full event

DEV is partnering to bring live events to the community. Join us or dismiss this billboard if you're not interested. ❤️