Skip to content
This repository has been archived by the owner on Aug 9, 2024. It is now read-only.

acntech-archive/fruit-classifier

Repository files navigation

Fruit classifier

Build Status codecov Python PEP8

This is repo is part of an assignment series. See assignments for details

This assignment is loosely based on this tutorial from pyimagesearch.

Usage

  1. Download the requirements in requirements.txt

  2. Download the chromedriver based on your operating system, and store it to the root directory of this repository

  3. Scrape images with python -m fruit_classifier.data_scraping

  4. Clean the data with python -m fruit_classifier.preprocessing

  5. Train with python -m fruit_classifier.train

  6. Predict with python -m fruit_classifier.predict -i <path_to_image>

    Unix-example (in windows replace / with \): python -m fruit_classifier.predict -i "test/data/raw/bananas/1. banana-1.png"