|
19 | 19 | "# TensorFlow Lite Object Detection API in Colab\n",
|
20 | 20 | "**Author:** Evan Juras, [EJ Technology Consultants](https://ejtech.io)\n",
|
21 | 21 | "\n",
|
22 |
| - "**Last updated:** 11/20/22\n", |
| 22 | + "**Last updated:** 1/15/22\n", |
23 | 23 | "\n",
|
24 | 24 | "**GitHub:** [TensorFlow Lite Object Detection](https://github.com/EdjeElectronics/TensorFlow-Lite-Object-Detection-on-Android-and-Raspberry-Pi)\n",
|
25 | 25 | "\n",
|
|
28 | 28 | "This notebook uses [the TensorFlow 2 Object Detection API](https://github.com/tensorflow/models/tree/master/research/object_detection) to train an SSD-MobileNet model or EfficientDet model with a custom dataset and convert it to TensorFlow Lite format. By working through this Colab, you'll be able to create and download a TFLite model that you can run on your PC, an Android phone, or an edge device like the Raspberry Pi.\n",
|
29 | 29 | "\n",
|
30 | 30 | "<p align=center>\n",
|
31 |
| - "<img src=\"https://s3.us-west-1.amazonaws.com/evanjuras.com/img/CoinDetectorDemo.gif\" height=\"350\"><br>\n", |
| 31 | + "<img src=\"https://s3.us-west-1.amazonaws.com/evanjuras.com/img/CoinDetectorDemo.gif\" height=\"360\"><br>\n", |
32 | 32 | "<i>Custom SSD-MobileNet-FPNLite model in action!</i>\n",
|
33 | 33 | "</p>\n",
|
34 | 34 | "\n",
|
35 |
| - "I made a YouTube video that walks through this guide step by step. I use a coin detection model as an example for the video. I recommend following along with the video while working through this notebook.\n", |
| 35 | + "I also made a YouTube video that walks through this guide step by step. I use a coin detection model as an example for the video. I recommend following along with the video while working through this notebook.\n", |
36 | 36 | "\n",
|
37 |
| - "*Link to video to be added here*\n", |
| 37 | + "<p align=center>\n", |
| 38 | + "<img src=\"https://s3.us-west-1.amazonaws.com/evanjuras.com/img/thumbnail-tflite-colab-1.png\" height=\"240\"><br>\n", |
| 39 | + "<a href=\"example.com\" target=\"_blank\"><i>Click here to go to the video!</i></a>\n", |
| 40 | + "</p>\n", |
38 | 41 | "\n",
|
39 | 42 | "**Important note: This notebook will be continuously updated to make sure it works with newer versions of TensorFlow. If you see any differences between the YouTube video and this notebook, always follow the notebook!**\n",
|
40 | 43 | "\n",
|
|
68 | 71 | "\n",
|
69 | 72 | "Watch the YouTube video below for instructions and tips on how to gather and label images for training an object detection model.\n",
|
70 | 73 | "\n",
|
71 |
| - "*Link to YouTube video*\n", |
| 74 | + "*Link to YouTube video will be added when it's ready!*\n", |
72 | 75 | "\n",
|
73 | 76 | "When you've finished gathering and labeling images, you should have a folder full of images and corresponding .xml data annotation file for each image. An example of a labeled image and the image folder for my coin detector model are shown below.\n",
|
74 | 77 | "\n",
|
|
445 | 448 | "\n",
|
446 | 449 | "Set the \"chosen_model\" variable to match the name of the model you'd like to train with. It's currently set to use the popular \"ssd-mobilenet-v2\" model. Click play on the next block once the chosen model has been set.\n",
|
447 | 450 | "\n",
|
448 |
| - "Not sure which model to pick? Check out my blog post comparing each model's speed and accuracy. *link to be added*" |
| 451 | + "Not sure which model to pick? [Check out my blog post comparing each model's speed and accuracy.](https://ejtech.io/learn/tflite-object-detection-model-comparison)" |
449 | 452 | ]
|
450 | 453 | },
|
451 | 454 | {
|
|
1657 | 1660 | "WoptFnAhCSrR",
|
1658 | 1661 | "5VI_Gh5dCd7w"
|
1659 | 1662 | ],
|
1660 |
| - "authorship_tag": "ABX9TyOdEG1NnBkg7Py/OPcq5IoY", |
| 1663 | + "authorship_tag": "ABX9TyPWkOZphlQaYIpfsw2iFLuk", |
1661 | 1664 | "include_colab_link": true
|
1662 | 1665 | },
|
1663 | 1666 | "gpuClass": "standard",
|
|
0 commit comments