magichampz commited on
Commit
3c10a49
1 Parent(s): 6e6e2d0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
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
  ![image/gif](https://cdn-uploads.huggingface.co/production/uploads/652dc3dab86e108d0fea458c/E7UZXLWPvU_39cxrF49jD.gif)
22
 
23
  ## Uses
24
- The files in the computer folder are meant for use on your own computer.
25
- You can create and train your own deep learning model using your own data and also test this model on your computer.
26
- The model was trained on Google colab, so create_training_data_array.py was used 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,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 Collab. 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
 
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
  ![image/gif](https://cdn-uploads.huggingface.co/production/uploads/652dc3dab86e108d0fea458c/E7UZXLWPvU_39cxrF49jD.gif)
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