Skip to content

Latest commit

 

History

History
51 lines (35 loc) · 1.4 KB

README.md

File metadata and controls

51 lines (35 loc) · 1.4 KB

Domino Classification with Python and Scikit-Learn

This repository contains a Python implementation for classifying dominos using Scikit-Learn.

Dataset validation / Inference result

Dataset validation Inference result

Installation

  1. Clone this repository:
$ git clone https://github.com/attilarepka/domino-classification-python-scikit-learn.git
  1. Navigate to the project directory:
$ cd domino-classification-python-scikit-learn
  1. Install the required Python packages:
$ pip install -r requirements.txt

Usage

  1. Dataset Preparation: Run dataset.py to split and create datasets from a large domino set image.
$ python dataset.py -i <input_image_path>

  1. Contour Validation and Display: Utilize dots.py for contour validation and display, which is used by dataset.py.

  2. Training and Validation: Use main.py for training and validation of the classifier.

$ python main.py [-o OUTPUT] [-d DATASET]
  1. Inferencing: Perform inferencing using predict.py based on the created dataset.

License

This project is licensed under the MIT License.