3.8 KiB
3.8 KiB
Tree Recogniser 7000
This repository contains all files for the Image recognition course of HU Electrical Engineering year 3.
Directories and files:
.
├── example/ (training assignments course)
├── out/
│ ├── img (images exported using CVSuite)/
│ │ └── (tag)_(preprocessor)_(date/time).png
│ ├── log (preprocessor export from CVSuite)/
│ │ └── result_(date/time).csv
│ └── models (exported OpenCV ML models for usage in CVSuite tests)/
│ └── model_(name).yaml
├── res/
│ ├── dataset (dataset for CVSuite, see README)/
│ │ ├── testing
│ │ ├── training
│ │ ├── validation
│ │ └── *.png
│ ├── essay/ (export photo's and graphs for report)
│ ├── trees/ (initial dataset)
│ └── *.png (photos required by assignments in .example/)
├── src/
│ ├── config/
│ │ └── config.json (CVSuite config, alter config.template.json to desired settings)
│ ├── experiments/ (standalone python scripts for experimentation)
│ ├── helpers/
│ │ ├── gui/
│ │ │ └── main.ui (pygubu ui configuration)
│ │ ├── test/ (ML test classes for CVSuite)
│ │ └── *.py (other CVSuite helper classes)
│ └── suite.py (CVSuite main script)
├── README.md (this file)
└── requirements.txt (pip install file)
How to:
Use the virtual environment
- Make sure you have the Python extension in VSCode
- Create a virtual environment using VSCode by entering the Command Palette, selecting "Python: Create Environment..." and choosing venv.
- VSCode will automatically include the venv in the integrated terminal, if you want to open it in another terminal, use the appropriate activation script in the
.venv
folder
$ ./.venv/Scripts/activate(.bat/.ps1)
- Install required packages using pip
$ pip install -r ./requirements.txt
Create a dataset
- Rename all images to include a tag and unique id, seperated by an underscore '_'
- e.g.
accasia_1210262
- e.g.
- Put all images into
./res/dataset
- Run the dataset tool:
$ python ./src/experiments/dataset.py
Run CVSuite (for the first time)
- Create
config.json
in the./src/config/
folder and copy the contents of the template - Edit
config.json
to fit your system, use full pathspath
should point to the dataset directorymodels
should point to trained ML models in YAML formatout
should point to the respective folders in the./out
foldersize
determines the display size in the suite
- Run CVSuite:
$ python ./src/suite.py
Train and export a KNN model
- Open CVSuite and select the desired training set
- Press 'Run analysis for entire dataset(!)', this will export a CSV file with all preprocessed data in the
./out
directory- Based on your system configuration, this might take a while
- Run the CVSuiteTestKNN CLI tool:
$ python ./src/helpers/test/knn.py -i ./out/result-(date/time).csv -o ./out/models/model_knn.yaml
- Edit your
config.json
to include the newly created model
Arne van Iterson
Tom Selier