https://github.com/xxczaki/ship-classifier

Classify ship types using machine learning (beta)
https://github.com/xxczaki/ship-classifier

artificial-intelligence classifier classify image-classification image-classifier image-recognition javascript machine-learning model nextjs photo react ship ships tensorflow tensorflow-examples tensorflow-js tensorflowjs

Last synced: 5 months ago
JSON representation

Classify ship types using machine learning (beta)

• Host: GitHub
• URL: https://github.com/xxczaki/ship-classifier
• Owner: xxczaki
• License: mit
• Created: 2019-11-09T22:42:27.000Z (about 6 years ago)
• Default Branch: master
• Last Pushed: 2025-06-05T18:20:17.000Z (7 months ago)
• Last Synced: 2025-06-06T03:02:33.326Z (7 months ago)
• Topics: artificial-intelligence, classifier, classify, image-classification, image-classifier, image-recognition, javascript, machine-learning, model, nextjs, photo, react, ship, ships, tensorflow, tensorflow-examples, tensorflow-js, tensorflowjs
• Language: JavaScript
• Homepage: https://ship-classifier.now.sh
• Size: 3.38 MB
• Stars: 2
• Watchers: 2
• Forks: 2
• Open Issues: 1
• Metadata Files:
  • Readme: readme.md
  • License: license.md

Awesome Lists containing this project

README

          
# Ship Classifier 🚢

> Classify ship types using machine learning (beta)

[![Build Status](https://travis-ci.org/xxczaki/ship-classifier.svg?branch=master)](https://travis-ci.org/xxczaki/ship-classifier)

[![XO code style](https://img.shields.io/badge/code_style-XO-5ed9c7.svg)](https://github.com/xojs/xo)

[![style: styled-components](https://img.shields.io/badge/style-%F0%9F%92%85%20styled--components-orange.svg?colorB=daa357&colorA=db748e)](https://github.com/styled-components/styled-components)

---

![Demo](demo.gif)

## About

This app uses a machine learning model trained with pictures of 8 different ship types:

- Passenger Vessel

- Container Vessel

- Bulk Vessel

- Roll-on/Roll-off Vessel

- Naval Vessel

- Chemical Tanker

- Sailing Ship

- Submarine

Each learning set contained between 50 and 78 images. The model itself was trained for about 1 minute.

To train the model I used [Teachable Machine](https://teachablemachine.withgoogle.com/), an AI Experiment by Google.

Ship Classifier uses [ml5](https://ml5js.org/), a high-level wrapper around [Tensorflow.js](https://www.tensorflow.org/js)

## Working with the model

If you want to try using the model in your own app, it's available [here](https://github.com/xxczaki/ship-classifier/tree/master/public/model).

In case you would like to train the model yourself, [here](https://ln2.sync.com/dl/956e0cd70/vxkbdr9r-c5cyjkt2-7f5vg3tt-7wqqfpdx) you can download learning sets with sample photos.

As I mentioned earlier, to train the model I used Teachable Machine with the following configuration:

```

Epochs: 150

Batch Size: 16

Learning Rate: 0.00135

```

## Development

> Hosted with [now Δ](https://zeit.com/now)

```

# Install dependencies

$ npm install

# Start in development mode

$ npm run dev

# Build for production

$ npm run build

```

## License

MIT