Skip to content

Commit 41d68b1

Browse files
author
Yijie Zhuang
committed
minor change to object detection notebook
1 parent aac4fd3 commit 41d68b1

File tree

2 files changed

+120
-41
lines changed

2 files changed

+120
-41
lines changed

‎introduction_to_amazon_algorithms/object_detection_pascalvoc_coco/object_detection_image_json_format.ipynb

Lines changed: 64 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,9 @@
7272
{
7373
"cell_type": "code",
7474
"execution_count": 2,
75-
"metadata": {},
75+
"metadata": {
76+
"collapsed": true
77+
},
7678
"outputs": [],
7779
"source": [
7880
"bucket = '<your_s3_bucket_name_here>' # custom bucket name.\n",
@@ -122,7 +124,9 @@
122124
{
123125
"cell_type": "code",
124126
"execution_count": null,
125-
"metadata": {},
127+
"metadata": {
128+
"collapsed": true
129+
},
126130
"outputs": [],
127131
"source": [
128132
"import os\n",
@@ -142,7 +146,9 @@
142146
{
143147
"cell_type": "code",
144148
"execution_count": null,
145-
"metadata": {},
149+
"metadata": {
150+
"collapsed": true
151+
},
146152
"outputs": [],
147153
"source": [
148154
"%%bash\n",
@@ -161,7 +167,9 @@
161167
{
162168
"cell_type": "code",
163169
"execution_count": null,
164-
"metadata": {},
170+
"metadata": {
171+
"collapsed": true
172+
},
165173
"outputs": [],
166174
"source": [
167175
"%%bash\n",
@@ -185,7 +193,9 @@
185193
{
186194
"cell_type": "code",
187195
"execution_count": null,
188-
"metadata": {},
196+
"metadata": {
197+
"collapsed": true
198+
},
189199
"outputs": [],
190200
"source": [
191201
"import json\n",
@@ -215,7 +225,9 @@
215225
{
216226
"cell_type": "code",
217227
"execution_count": null,
218-
"metadata": {},
228+
"metadata": {
229+
"collapsed": true
230+
},
219231
"outputs": [],
220232
"source": [
221233
"def get_mapper_fn(map): \n",
@@ -236,7 +248,9 @@
236248
{
237249
"cell_type": "code",
238250
"execution_count": null,
239-
"metadata": {},
251+
"metadata": {
252+
"collapsed": true
253+
},
240254
"outputs": [],
241255
"source": [
242256
"file_name = './annotations/instances_val2017.json'\n",
@@ -284,7 +298,9 @@
284298
{
285299
"cell_type": "code",
286300
"execution_count": null,
287-
"metadata": {},
301+
"metadata": {
302+
"collapsed": true
303+
},
288304
"outputs": [],
289305
"source": [
290306
"import os\n",
@@ -304,7 +320,9 @@
304320
{
305321
"cell_type": "code",
306322
"execution_count": null,
307-
"metadata": {},
323+
"metadata": {
324+
"collapsed": true
325+
},
308326
"outputs": [],
309327
"source": [
310328
"import shutil\n",
@@ -336,7 +354,9 @@
336354
{
337355
"cell_type": "code",
338356
"execution_count": null,
339-
"metadata": {},
357+
"metadata": {
358+
"collapsed": true
359+
},
340360
"outputs": [],
341361
"source": [
342362
"%%time\n",
@@ -367,7 +387,9 @@
367387
{
368388
"cell_type": "code",
369389
"execution_count": null,
370-
"metadata": {},
390+
"metadata": {
391+
"collapsed": true
392+
},
371393
"outputs": [],
372394
"source": [
373395
"s3_output_location = 's3://{}/{}/output'.format(bucket, prefix)"
@@ -384,7 +406,9 @@
384406
{
385407
"cell_type": "code",
386408
"execution_count": null,
387-
"metadata": {},
409+
"metadata": {
410+
"collapsed": true
411+
},
388412
"outputs": [],
389413
"source": [
390414
"od_model = sagemaker.estimator.Estimator(training_image,\n",
@@ -410,7 +434,9 @@
410434
{
411435
"cell_type": "code",
412436
"execution_count": null,
413-
"metadata": {},
437+
"metadata": {
438+
"collapsed": true
439+
},
414440
"outputs": [],
415441
"source": [
416442
"od_model.set_hyperparameters(base_network='resnet-50',\n",
@@ -441,7 +467,9 @@
441467
{
442468
"cell_type": "code",
443469
"execution_count": null,
444-
"metadata": {},
470+
"metadata": {
471+
"collapsed": true
472+
},
445473
"outputs": [],
446474
"source": [
447475
"train_data = sagemaker.session.s3_input(s3_train_data, distribution='FullyReplicated', \n",
@@ -469,7 +497,9 @@
469497
{
470498
"cell_type": "code",
471499
"execution_count": null,
472-
"metadata": {},
500+
"metadata": {
501+
"collapsed": true
502+
},
473503
"outputs": [],
474504
"source": [
475505
"od_model.fit(inputs=data_channels, logs=True)"
@@ -486,7 +516,9 @@
486516
{
487517
"cell_type": "code",
488518
"execution_count": null,
489-
"metadata": {},
519+
"metadata": {
520+
"collapsed": true
521+
},
490522
"outputs": [],
491523
"source": [
492524
"object_detector = od_model.deploy(initial_instance_count = 1,\n",
@@ -504,7 +536,9 @@
504536
{
505537
"cell_type": "code",
506538
"execution_count": null,
507-
"metadata": {},
539+
"metadata": {
540+
"collapsed": true
541+
},
508542
"outputs": [],
509543
"source": [
510544
"!wget -O test.jpg https://images.pexels.com/photos/980382/pexels-photo-980382.jpeg\n",
@@ -527,7 +561,9 @@
527561
{
528562
"cell_type": "code",
529563
"execution_count": null,
530-
"metadata": {},
564+
"metadata": {
565+
"collapsed": true
566+
},
531567
"outputs": [],
532568
"source": [
533569
"import json\n",
@@ -550,7 +586,9 @@
550586
{
551587
"cell_type": "code",
552588
"execution_count": null,
553-
"metadata": {},
589+
"metadata": {
590+
"collapsed": true
591+
},
554592
"outputs": [],
555593
"source": [
556594
"def visualize_detection(img_file, dets, classes=[], thresh=0.6):\n",
@@ -613,7 +651,9 @@
613651
{
614652
"cell_type": "code",
615653
"execution_count": null,
616-
"metadata": {},
654+
"metadata": {
655+
"collapsed": true
656+
},
617657
"outputs": [],
618658
"source": [
619659
"object_categories = ['person', 'bicycle', 'car', 'motorbike', 'aeroplane', 'bus', 'train', 'truck', 'boat', \n",
@@ -630,7 +670,7 @@
630670
"threshold = 0.20\n",
631671
"\n",
632672
"# Visualize the detections.\n",
633-
"visualize_detection(file_name, detections, object_categories, threshold)"
673+
"visualize_detection(file_name, detections['prediction'], object_categories, threshold)"
634674
]
635675
},
636676
{
@@ -644,7 +684,9 @@
644684
{
645685
"cell_type": "code",
646686
"execution_count": null,
647-
"metadata": {},
687+
"metadata": {
688+
"collapsed": true
689+
},
648690
"outputs": [],
649691
"source": [
650692
"sagemaker.Session().delete_endpoint(object_detector.endpoint)"

0 commit comments

Comments
 (0)