Commit
·
3c10a49
1
Parent(s):
6e6e2d0
Update README.md
Browse files
README.md
CHANGED
@@ -13,7 +13,7 @@ Achieved a 93% validation accuracy
|
|
13 |
|
14 |
- **Developed by:** Aveek Goswami, Amos Koh
|
15 |
- **Funded by [optional]:** Nullspace Robotics Singapore
|
16 |
-
- **Model type:** Convolutional Neural Network
|
17 |
|
18 |
### Model Sources
|
19 |
|
@@ -21,9 +21,9 @@ Achieved a 93% validation accuracy
|
|
21 |

|
22 |
|
23 |
## Uses
|
24 |
-
The files in the
|
25 |
-
You can
|
26 |
-
The model was trained on Google
|
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
|
@@ -41,7 +41,7 @@ More images can be taken by editing the motion_detection_and_image_classificatio
|
|
41 |
<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
|
42 |
|
43 |
### Training Procedure
|
44 |
-
The model was trained using the GPU's available on Google
|
45 |
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
46 |
|
47 |
#### Preprocessing
|
|
|
13 |
|
14 |
- **Developed by:** Aveek Goswami, Amos Koh
|
15 |
- **Funded by [optional]:** Nullspace Robotics Singapore
|
16 |
+
- **Model type:** Convolutional Neural Network (CNN)
|
17 |
|
18 |
### Model Sources
|
19 |
|
|
|
21 |

|
22 |
|
23 |
## Uses
|
24 |
+
The files in the create-model folder are meant to be used on your own computer.
|
25 |
+
You can train your own deep learning model using your own data and test this model on your computer using testing-tflite-model.py on a single image.
|
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
|
|
|
41 |
<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
|
42 |
|
43 |
### Training Procedure
|
44 |
+
The model was trained using the GPU's available on Google Colab. The jupyter notebook loaded the data from a npy file (in the dataset card), which contained all the images as well as their category labels.
|
45 |
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
46 |
|
47 |
#### Preprocessing
|