/manhole-cover-model

Primary LanguagePython

Manhole Cover Classifier

ezgif-4-b657bbd0b1 ezgif-5-35c28f41eb ezgif-4-1abd2a1947

This repository predicts the class of a Manhole Cover image from 12 different classes:

  • Rost/Strassenrost
  • Vollguss/Pickelloch belueftet
  • Gussbeton/Pickelloch geschlossen
  • Vollguss/Pickelloch geschlossen
  • Gussbeton/Pickelloch belueftet
  • Vollguss/Handgriff geschlossen
  • Gussbeton/Handgriff seitlich
  • Rost/Einlauf rund
  • Rost/Strassenrost gewoelbt
  • Vollguss/Aufklappbar
  • Gussbeton/Handgriff mitte
  • Vollguss/Handgriff geschlossen, verschraubt

Installation

  1. Clone project locally
git clone git@github.com:FiratSaritas/manhole-cover-model.git

  1. Download model from Google Drive and add it to the folder ./model here: https://drive.google.com/file/d/1JYONs6lFC2mbSC2KOwpnGopcbL1OTXBc/view?usp=sharing

  2. Install required packages

Preferably you create a new enviorment (conda environment is also possible).

pip install -r requirements.txt

Usage

python run.py [Url of image]

Output:

  • result.json-file in the current folder

Additional Usage with parameter:

python run.py --help

Examples

python run.py image.png

or with number of top predictions:

python run.py image.png 3