added template extraction to readme

This commit is contained in:
Tom Selier 2023-10-22 18:59:50 +02:00
parent 58e521cb6e
commit 0bd7b7809a

View File

@ -67,6 +67,8 @@ $ pip install -e .
```bash ```bash
$ python ./src/experiments/dataset.py $ python ./src/experiments/dataset.py
``` ```
4. (optional) run the template extraction tool
5. (optional) run the dataset splitter tool
### Run CVSuite (for the first time) ### Run CVSuite (for the first time)
1. Create `config.json` in the `./src/config/` folder and copy the contents of the template 1. Create `config.json` in the `./src/config/` folder and copy the contents of the template
@ -103,11 +105,23 @@ $ python ./src/helpers/test/knn.py -i ./out/result-(date/time).csv -o ./out/mode
- `-m` Model to train; `dectree`, `randforest` or `extratree` - `-m` Model to train; `dectree`, `randforest` or `extratree`
- `-s` Scaler file to use (`.pkl` file) - `-s` Scaler file to use (`.pkl` file)
```sh ```sh
python ./src/helpers/test/decision_tree.py -i ./out/result-(date/time).csv -o ./out/models/ -m 'dectree' -s ./out/models/scale_(date/time).pkl $ python ./src/helpers/test/decision_tree.py -i ./out/result-(date/time).csv -o ./out/models/ -m 'dectree' -s ./out/models/scale_(date/time).pkl
``` ```
2. The script generates one `.pkl` file based on the chosen model 2. The script generates one `.pkl` file based on the chosen model
3. Edit your `config.json` to include the newly created model 3. Edit your `config.json` to include the newly created model
### Template extraction
> :warning: **Please note:** <br>
> This tool uses the legacy format for datasets.<br>
> Images are sorted using folders, instead of by name
1. Images should have four standard Aruco markers clearly visible
2. Run the template extraction tool with an input directory as argument
```sh
$ python ./src/experiments/template_extraction/script.py ./dataset
```
3. The script generates new folders, ending with `_out`
4. The paths to any failed images are saved in `skipped.txt`
--- ---
Arne van Iterson<br> Arne van Iterson<br>