|
55 | 55 | {
|
56 | 56 | "cell_type": "code",
|
57 | 57 | "execution_count": 2,
|
58 |
| - "metadata": { |
59 |
| - "scrolled": false |
60 |
| - }, |
| 58 | + "metadata": {}, |
61 | 59 | "outputs": [],
|
62 | 60 | "source": [
|
63 | 61 | "import json\n",
|
|
195 | 193 | "source": [
|
196 | 194 | "client = boto3.client('s3')\n",
|
197 | 195 | "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", |
199 | 197 | " MaxKeys=20)\n",
|
200 | 198 | "scenes = [s['Key'] for s in scenes['Contents'] if s['Key'].endswith('.tif')][0:10]"
|
201 | 199 | ]
|
|
291 | 289 | "source": [
|
292 | 290 | "def item_to_labels_url(item):\n",
|
293 | 291 | " 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", |
296 | 294 | " return(join(d, b))"
|
297 | 295 | ]
|
298 | 296 | },
|
|
314 | 312 | " params = {\n",
|
315 | 313 | " 'id': item.id + '-labels',\n",
|
316 | 314 | " 'bbox': item.bbox,\n",
|
| 315 | + " 'geometry': box(*item.bbox).__geo_interface__,\n", |
317 | 316 | " 'datetime': item.datetime,\n",
|
318 | 317 | " 'properties': {},\n",
|
319 | 318 | " 'label_description': 'Building labels for scene {}'.format(item.id),\n",
|
320 | 319 | " 'label_type': 'vector',\n",
|
321 | 320 | " 'label_properties': 'partialBuilding'\n",
|
322 | 321 | " }\n",
|
323 | 322 | " \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", |
328 | 323 | " label_item = LabelItem(**params)\n",
|
329 | 324 | " \n",
|
330 | 325 | " label_item.add_asset(key='label', asset=Asset(href=label_uri, media_type='application/geo+json'))\n",
|
|
363 | 358 | "name": "stdout",
|
364 | 359 | "output_type": "stream",
|
365 | 360 | "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" |
376 | 371 | ]
|
377 | 372 | }
|
378 | 373 | ],
|
|
392 | 387 | "text": [
|
393 | 388 | "* <Catalog id=spacenet>\n",
|
394 | 389 | " * <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" |
415 | 410 | ]
|
416 | 411 | }
|
417 | 412 | ],
|
|
438 | 433 | ],
|
439 | 434 | "metadata": {
|
440 | 435 | "kernelspec": {
|
441 |
| - "display_name": "Python [default]", |
| 436 | + "display_name": "Python 3", |
442 | 437 | "language": "python",
|
443 | 438 | "name": "python3"
|
444 | 439 | },
|
|
452 | 447 | "name": "python",
|
453 | 448 | "nbconvert_exporter": "python",
|
454 | 449 | "pygments_lexer": "ipython3",
|
455 |
| - "version": "3.6.5" |
| 450 | + "version": "3.6.7" |
456 | 451 | }
|
457 | 452 | },
|
458 | 453 | "nbformat": 4,
|
459 |
| - "nbformat_minor": 2 |
| 454 | + "nbformat_minor": 4 |
460 | 455 | }
|
0 commit comments