/ml-image-classification-for-beginner

Primary LanguagePython

Image classification

Current repository is based on Image Classification for beginner tutorial (with some refactoring).

This repository will also be the basis for the "Workshop 1. Computer Vision | Image classification".

How to setup

  1. Install Miniconda for your operation system. Current repository using conda 24.1.0
  2. Clone this repository to your workshop folder
  3. Run next command to create an environment from an environment.yml file
conda env create -y -f environment.yml

Verify that the new environment was installed correctly:

conda env list
  1. Activate a new environment
conda activate cv-workshop-01-3_11

  1. Create data subfolder in the project folder. Download the data set - Rice Image Dataset and place it in the data folder.

  2. Train the model

python ./src/image_classification_for_beginner/main.py
  1. Test the model with selected image
python ./src/image_classification_for_beginner/test_model.py