Skip to content

Commit dd13790

Browse files
authored
Merge pull request robmarkcole#19 from robmarkcole/discuss-custom-models
Update README.md
2 parents 9dd15e3 + d1f7e36 commit dd13790

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

README.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,9 @@ pip3 install -r requirements.txt
1313
```
1414

1515
## Models
16-
For convenience a couple of models are included in this repo and used by default. A description of each model is included in its directory. Additional models are available [here](https://github.com/google-coral/edgetpu/tree/master/test_data)
16+
For convenience a couple of models are included in this repo and used by default. A description of each model is included in its directory. Additional models are available [here](https://github.com/google-coral/edgetpu/tree/master/test_data).
17+
18+
If you want to create custom models, there is the easy way, and the longer but more flexible way. The easy way is to use [teachablemachine](https://teachablemachine.withgoogle.com/train/image), which I have done in this repo for the dogs-vs-cats model. This is limited to image classification but is very straightforward. The longer way allows you to use any neural network architecture to produce a tensorflow model, which you then convert to a tflite model. An example of this approch is described in [this article](https://towardsdatascience.com/inferences-from-a-tf-lite-model-transfer-learning-on-a-pre-trained-model-e16e7c5f0ee6), or jump straight [to the code](https://github.com/arshren/TFLite/blob/master/Transfer%20Learning%20with%20TFLite-Copy1.ipynb).
1719

1820
## Usage
1921
Start the server on port 5000 (default is port 5000):

0 commit comments

Comments
 (0)