Skip to content

Commit 7170eab

Browse files
authored
Merge pull request stac-utils#48 from nrweir/fix_spacenet_demo
fixing for new sn2 s3 structure
2 parents c6e5a95 + f142945 commit 7170eab

File tree

1 file changed

+38
-43
lines changed

1 file changed

+38
-43
lines changed

docs/tutorials/pystac-spacenet-tutorial.ipynb

Lines changed: 38 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,7 @@
5555
{
5656
"cell_type": "code",
5757
"execution_count": 2,
58-
"metadata": {
59-
"scrolled": false
60-
},
58+
"metadata": {},
6159
"outputs": [],
6260
"source": [
6361
"import json\n",
@@ -195,7 +193,7 @@
195193
"source": [
196194
"client = boto3.client('s3')\n",
197195
"scenes = client.list_objects(Bucket='spacenet-dataset', \n",
198-
" Prefix='SpaceNet_Buildings_Dataset_Round2/spacenetV2_Train/AOI_2_Vegas/RGB-PanSharpen/',\n",
196+
" Prefix='spacenet/SN2_buildings/train/AOI_2_Vegas/PS-RGB/',\n",
199197
" MaxKeys=20)\n",
200198
"scenes = [s['Key'] for s in scenes['Contents'] if s['Key'].endswith('.tif')][0:10]"
201199
]
@@ -291,8 +289,8 @@
291289
"source": [
292290
"def item_to_labels_url(item):\n",
293291
" image_uri = item.assets['image'].href\n",
294-
" d = dirname(image_uri).replace('RGB-PanSharpen', 'geojson/buildings')\n",
295-
" b = basename(image_uri).replace('RGB-PanSharpen', 'buildings').replace('.tif', '.geojson')\n",
292+
" d = dirname(image_uri).replace('PS-RGB', 'geojson_buildings')\n",
293+
" b = basename(image_uri).replace('PS-RGB', 'geojson_buildings').replace('.tif', '.geojson')\n",
296294
" return(join(d, b))"
297295
]
298296
},
@@ -314,17 +312,14 @@
314312
" params = {\n",
315313
" 'id': item.id + '-labels',\n",
316314
" 'bbox': item.bbox,\n",
315+
" 'geometry': box(*item.bbox).__geo_interface__,\n",
317316
" 'datetime': item.datetime,\n",
318317
" 'properties': {},\n",
319318
" 'label_description': 'Building labels for scene {}'.format(item.id),\n",
320319
" 'label_type': 'vector',\n",
321320
" 'label_properties': 'partialBuilding'\n",
322321
" }\n",
323322
" \n",
324-
" with open(label_uri) as gj:\n",
325-
" features = json.load(gj)\n",
326-
" geom_collection = GeometryCollection([shape(feature['geometry']) for feature in features['features']])\n",
327-
" params['geometry'] = geom_collection.__geo_interface__\n",
328323
" label_item = LabelItem(**params)\n",
329324
" \n",
330325
" label_item.add_asset(key='label', asset=Asset(href=label_uri, media_type='application/geo+json'))\n",
@@ -363,16 +358,16 @@
363358
"name": "stdout",
364359
"output_type": "stream",
365360
"text": [
366-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1\n",
367-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img10\n",
368-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1002\n",
369-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1003\n",
370-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1004\n",
371-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1006\n",
372-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1007\n",
373-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1009\n",
374-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img101\n",
375-
"Completed item: RGB-PanSharpen_AOI_2_Vegas_img1010\n"
361+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1\n",
362+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img10\n",
363+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1002\n",
364+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1003\n",
365+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1004\n",
366+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1006\n",
367+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1007\n",
368+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1009\n",
369+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img101\n",
370+
"Completed item: SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1010\n"
376371
]
377372
}
378373
],
@@ -392,26 +387,26 @@
392387
"text": [
393388
"* <Catalog id=spacenet>\n",
394389
" * <Collection id=vegas>\n",
395-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1>\n",
396-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1-labels>\n",
397-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img10>\n",
398-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img10-labels>\n",
399-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1002>\n",
400-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1002-labels>\n",
401-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1003>\n",
402-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1003-labels>\n",
403-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1004>\n",
404-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1004-labels>\n",
405-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1006>\n",
406-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1006-labels>\n",
407-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1007>\n",
408-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1007-labels>\n",
409-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1009>\n",
410-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1009-labels>\n",
411-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img101>\n",
412-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img101-labels>\n",
413-
" * <Item id=RGB-PanSharpen_AOI_2_Vegas_img1010>\n",
414-
" * <LabelItem id=RGB-PanSharpen_AOI_2_Vegas_img1010-labels>\n"
390+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1>\n",
391+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1-labels>\n",
392+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img10>\n",
393+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img10-labels>\n",
394+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1002>\n",
395+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1002-labels>\n",
396+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1003>\n",
397+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1003-labels>\n",
398+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1004>\n",
399+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1004-labels>\n",
400+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1006>\n",
401+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1006-labels>\n",
402+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1007>\n",
403+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1007-labels>\n",
404+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1009>\n",
405+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1009-labels>\n",
406+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img101>\n",
407+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img101-labels>\n",
408+
" * <Item id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1010>\n",
409+
" * <LabelItem id=SN2_buildings_train_AOI_2_Vegas_PS-RGB_img1010-labels>\n"
415410
]
416411
}
417412
],
@@ -438,7 +433,7 @@
438433
],
439434
"metadata": {
440435
"kernelspec": {
441-
"display_name": "Python [default]",
436+
"display_name": "Python 3",
442437
"language": "python",
443438
"name": "python3"
444439
},
@@ -452,9 +447,9 @@
452447
"name": "python",
453448
"nbconvert_exporter": "python",
454449
"pygments_lexer": "ipython3",
455-
"version": "3.6.5"
450+
"version": "3.6.7"
456451
}
457452
},
458453
"nbformat": 4,
459-
"nbformat_minor": 2
454+
"nbformat_minor": 4
460455
}

0 commit comments

Comments
 (0)