Commit
·
1e19a95
1
Parent(s):
3c10a49
Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ Achieved a 93% validation accuracy
|
|
| 18 |
### Model Sources
|
| 19 |
|
| 20 |
- **Repository:** https://github.com/magichampz/lego-sorting-machine-ag-ak
|
| 21 |
-
|
| 22 |
|
| 23 |
## Uses
|
| 24 |
The files in the create-model folder are meant to be used on your own computer.
|
|
@@ -26,6 +26,10 @@ You can train your own deep learning model using your own data and test this mo
|
|
| 26 |
The model was trained on Google Colab, so create_training_data_array.py was used to creata a numpy array file to upload data in the form of a numpy array to Google Colab.
|
| 27 |
After transfering the tflite model to your Pi, you can then run the image classification file in the raspberry-pi folder to detect and classify lego pieces in real time.
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
## Bias, Limitations and Recommendations
|
| 30 |
The images of the lego pieces used to train the model were taken in room lighting conditions, illuminated with a torchlight. <br>
|
| 31 |
To use the model, would recommend trying to recreate the conditions and achieve photographs with a similar lighting. <br>
|
|
|
|
| 18 |
### Model Sources
|
| 19 |
|
| 20 |
- **Repository:** https://github.com/magichampz/lego-sorting-machine-ag-ak
|
| 21 |
+
|
| 22 |
|
| 23 |
## Uses
|
| 24 |
The files in the create-model folder are meant to be used on your own computer.
|
|
|
|
| 26 |
The model was trained on Google Colab, so create_training_data_array.py was used to creata a numpy array file to upload data in the form of a numpy array to Google Colab.
|
| 27 |
After transfering the tflite model to your Pi, you can then run the image classification file in the raspberry-pi folder to detect and classify lego pieces in real time.
|
| 28 |
|
| 29 |
+
Example of real time object detection and classification:
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
|
| 33 |
## Bias, Limitations and Recommendations
|
| 34 |
The images of the lego pieces used to train the model were taken in room lighting conditions, illuminated with a torchlight. <br>
|
| 35 |
To use the model, would recommend trying to recreate the conditions and achieve photographs with a similar lighting. <br>
|