Skip to content

Commit d132ae4

Browse files
minor cleanups
1 parent 4a3dc81 commit d132ae4

File tree

1 file changed

+16
-18
lines changed

1 file changed

+16
-18
lines changed

Train_TFLite2_Object_Detction_Model.ipynb

+16-18
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@
301301
"source": [
302302
"!mkdir /content/images\n",
303303
"!unzip -q images.zip -d /content/images/all\n",
304-
"!mkdir /content/images/train; mkdir /content/images/validation; mkdir /content/images/test\n"
304+
"!mkdir /content/images/train; mkdir /content/images/validation; mkdir /content/images/test"
305305
]
306306
},
307307
{
@@ -512,9 +512,6 @@
512512
"# Download pre-trained model weights\n",
513513
"import tarfile\n",
514514
"download_tar = 'http://download.tensorflow.org/models/object_detection/tf2/20200711/' + pretrained_checkpoint\n",
515-
"if chosen_model == 'centernet-mobilenet-v2':\n",
516-
" download_tar = 'http://download.tensorflow.org/models/object_detection/tf2/20210210/' + pretrained_checkpoint\n",
517-
"\n",
518515
"!wget {download_tar}\n",
519516
"tar = tarfile.open(pretrained_checkpoint)\n",
520517
"tar.extractall()\n",
@@ -982,8 +979,7 @@
982979
" for detection in detections:\n",
983980
" f.write('%s %.4f %d %d %d %d\\n' % (detection[0], detection[1], detection[2], detection[3], detection[4], detection[5]))\n",
984981
"\n",
985-
" return\n",
986-
"\n"
982+
" return"
987983
]
988984
},
989985
{
@@ -1142,17 +1138,7 @@
11421138
{
11431139
"cell_type": "code",
11441140
"source": [
1145-
"cd /content/mAP"
1146-
],
1147-
"metadata": {
1148-
"id": "q1rIsyscS-42"
1149-
},
1150-
"execution_count": null,
1151-
"outputs": []
1152-
},
1153-
{
1154-
"cell_type": "code",
1155-
"source": [
1141+
"%cd /content/mAP\n",
11561142
"!python calculate_map_cartucho.py --labels=/content/labelmap.txt"
11571143
],
11581144
"metadata": {
@@ -1659,7 +1645,19 @@
16591645
"colab": {
16601646
"provenance": [],
16611647
"toc_visible": true,
1662-
"authorship_tag": "ABX9TyM3+52g8zcyCZc+O2wbS+Te",
1648+
"collapsed_sections": [
1649+
"4VAvZo8qE4u5",
1650+
"sxb8_h-QFErO",
1651+
"eydREUsMGUUR",
1652+
"eGEUZYAMEZ6f",
1653+
"-19zML6oEO7l",
1654+
"kPg8oMnQDYKl",
1655+
"RDQrtQhvC3oG",
1656+
"5i40ve0SCLaE",
1657+
"WoptFnAhCSrR",
1658+
"5VI_Gh5dCd7w"
1659+
],
1660+
"authorship_tag": "ABX9TyObWrM2Q3JQAEGGYQZA6Scy",
16631661
"include_colab_link": true
16641662
},
16651663
"gpuClass": "standard",

0 commit comments

Comments
 (0)