Skip to content

Commit 7950b94

Browse files
Update readme and notebook
1 parent 4754b0b commit 7950b94

2 files changed

Lines changed: 197 additions & 312 deletions

File tree

‎README.md‎

Lines changed: 20 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
![EPFL Center for Imaging logo](https://imaging.epfl.ch/resources/logo-for-gitlab.svg)
22
# Introduction to Machine Learning for Vision Applications
33

4-
This workshop introduces [...]. It walks through:
4+
This workshop takes a hands-on approach to exploring some of the fundamental concepts of machine learning applied to computer vision. It walks through:
55

6-
- ...
7-
- ...
8-
- ...
6+
- Using Marimo
7+
- Working with an image dataset
8+
- Training an image classifier
9+
- Comparing different models
10+
- Running a model live on a camera feed
911

1012
## Quick start
1113

@@ -31,10 +33,21 @@ You should be able to access the notebook locally at: http://localhost:2718/.
3133

3234
## Scripts
3335

34-
To reproduce..., run the scripts:
36+
**Acquiring a training set of images**
3537

36-
- [`preprocessing.py`]() => Run on a folder with `.jpg` captures from the microscope. Will save a `digits` subfolder with extracted digits.
37-
- [`inference.py`]() => Run live using a saved `pipeline.pkl` on a camera device.
38+
To extract a training dataset of labelled digits from images captured with a camera device, use:
39+
40+
- [`preprocessing.py`](./preprocessing.py)
41+
42+
This script will save the training dataset in a local `dataset` folder. The class labels are inferred from the predetermined positions of the digits in a 3 X 4 grid.
43+
44+
**Testing the model live on a camera**
45+
46+
To extract digits and classify them based on the trained Scikit-learn pipeline, use:
47+
48+
- [`inference.py`](./inference.py)
49+
50+
This script will reload a Scikit-learn `pipeline.pkl` file, run the pipeline on the video feed from a camera device and display the results.
3851

3952
## License
4053

0 commit comments

Comments
 (0)