Skip to content
This repository was archived by the owner on May 1, 2025. It is now read-only.

Commit 7ee0b3a

Browse files
committed
Add links to new detection tutorials and update notebook names (HTML titles)
PiperOrigin-RevId: 377979638
1 parent 38f3727 commit 7ee0b3a

7 files changed

Lines changed: 36 additions & 20 deletions

README.md

Lines changed: 24 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -7,26 +7,38 @@ Colab, using the links below.
77

88
Train and build a model for the Edge TPU:
99

10-
+ [Retrain a classification model using post-training quantization (with TF2)](https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_classification_ptq_tf2.ipynb)
10+
+ [Retrain a classification model using post-training quant (TF2)](https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_classification_ptq_tf2.ipynb)
1111

12-
This shows how to build an image classification model with Keras, train it
13-
with a custom dataset, quantize it with post-training quantization, and then
14-
compile it for the Edge TPU.
12+
This notebook builds the MobileNet V2 classification model using Keras
13+
(with TF2), retrains it to classify different flowers, quantizes it with post-
14+
training quantization, and then compiles it for the Edge TPU.
1515

1616
We have another [version of this tutorial using TF1](
1717
https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_classification_ptq_tf1.ipynb).
1818

19-
+ [Retrain a classification model using quant-aware training (with TF1)](https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_classification_qat_tf1.ipynb)
19+
+ [Retrain a classification model using quant-aware training (TF1)](https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_classification_qat_tf1.ipynb)
2020

21-
This shows how to retrain a quant-aware classification model using TF1. This
22-
process is complex, so the notebook calls upon several Python scripts
23-
to do the work (links to these scripts are provided in the notebook).
21+
This notebook retrains a quant-aware version of MobileNet V1 (with TF1) to
22+
classify different flowers, and then compiles it for the Edge TPU.
2423

25-
+ [Retrain an object model using quant-aware training (with TF1)](https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_detection_qat_tf1.ipynb)
24+
+ [Retrain an object detection model (MobileNet) using quant-aware training (TF1)](https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_detection_qat_tf1.ipynb)
25+
26+
This notebook retrains a quant-aware version of MobileNet V1 SSD (with TF1)
27+
to detect cats and dogs, and then compiles it for the Edge TPU.
28+
29+
+ [Retrain an object detection model (MobileDet) using post-training quant (TF1)](https://colab.research.google.com/github/google-coral/tutorials/blob/master/retrain_ssdlite_mobiledet_qat_tf1.ipynb)
30+
31+
This notebook retrains the SSDLite MobileDet model to detect cats and dogs,
32+
quantizes it with post-training quantization, and then compiles it for the
33+
Edge TPU.
34+
35+
+ [Retrain an object detection model (EfficientDet) using TF Lite Model Maker (TF2)](https://github.com/google-coral/tutorials/blob/master/retrain_efficientdet_model_maker_tf2.ipynb)
36+
37+
This notebook retrains the EfficientDet-Lite model using the TensorFlow Lite
38+
Model Maker library (based on TF2) to detect different salad ingredients,
39+
exports it to TF Lite (using post-training quantization), and then compiles it
40+
for the Edge TPU.
2641

27-
This shows how to retrain a quant-aware object detection model using TF1. This
28-
process is complex, so the notebook calls upon several Python scripts
29-
to do the work (links to these scripts are provided in the notebook).
3042

3143
Other tutorials:
3244

retrain_classification_ptq_tf1.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1133,7 +1133,7 @@
11331133
"collapsed_sections": [
11341134
"license"
11351135
],
1136-
"name": "retrain_classification_ptq_tf1.ipynb",
1136+
"name": "Retrain MobileNet V2 classifier for the Edge TPU (TF1)",
11371137
"private_outputs": true,
11381138
"provenance": [],
11391139
"toc_visible": true

retrain_classification_ptq_tf2.ipynb

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1112,10 +1112,13 @@
11121112
"metadata": {
11131113
"accelerator": "GPU",
11141114
"colab": {
1115-
"collapsed_sections": [],
1116-
"name": "retrain_classification_ptq_tf2.ipynb",
1115+
"collapsed_sections": [
1116+
"license"
1117+
],
1118+
"name": "Retrain MobileNet V2 classifier for the Edge TPU (TF2)",
11171119
"private_outputs": true,
1118-
"provenance": []
1120+
"provenance": [],
1121+
"toc_visible": true
11191122
},
11201123
"kernelspec": {
11211124
"display_name": "Python 3",

retrain_classification_qat_tf1.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,7 @@
395395
"collapsed_sections": [
396396
"license"
397397
],
398-
"name": "retrain_classification_qat_tf1.ipynb",
398+
"name": "Retrain MobileNet V1 classifier for the Edge TPU (TF1, quant-aware)",
399399
"private_outputs": true,
400400
"provenance": [],
401401
"toc_visible": true

retrain_detection_qat_tf1.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -494,7 +494,7 @@
494494
"collapsed_sections": [
495495
"license"
496496
],
497-
"name": "retrain_detection_qat_tf1.ipynb",
497+
"name": "Retrain SSD MobileNet V1 detector for the Edge TPU (TF1)",
498498
"private_outputs": true,
499499
"provenance": [],
500500
"toc_visible": true

retrain_efficientdet_model_maker_tf2.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -618,7 +618,8 @@
618618
"collapsed_sections": [
619619
"license"
620620
],
621-
"name": "Retrain EfficientDet for the Edge TPU",
621+
"name": "Retrain EfficientDet-Lite detector for the Edge TPU (TF2)",
622+
"private_outputs": true,
622623
"provenance": [],
623624
"toc_visible": true
624625
},

retrain_ssdlite_mobiledet_qat_tf1.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -953,7 +953,7 @@
953953
"license"
954954
],
955955
"machine_shape": "hm",
956-
"name": "retrain_ssdlite_mobiledet_qat_tf1.ipynb",
956+
"name": "Retrain SSDLite MobileDet detector for the Edge TPU (TF1)",
957957
"private_outputs": true,
958958
"provenance": [],
959959
"toc_visible": true

0 commit comments

Comments
 (0)