|
6 | 6 | from subprocess import run, Popen, PIPE |
7 | 7 | import json |
8 | 8 | from functools import partial |
9 | | -import datetime |
10 | 9 |
|
11 | 10 | import numpy as np |
12 | 11 | import mapbox_vector_tile |
13 | 12 | import pyproj |
14 | | -import pandas as pd |
15 | 13 | from shapely.geometry import shape, mapping, Polygon |
16 | 14 | from shapely.errors import TopologicalError |
17 | 15 | from rasterio.features import rasterize |
18 | 16 | from geojson import Feature, FeatureCollection as fc |
19 | | -from pandas.io.json import json_normalize |
20 | 17 | from mercantile import tiles, feature, Tile |
21 | 18 | from PIL import Image, ImageDraw |
22 | 19 | from tilepie import tilereduce |
@@ -74,7 +71,7 @@ def make_labels(dest_folder, zoom, country, classes, ml_type, bounding_box, spar |
74 | 71 | mbtiles_file_zoomed = op.join(dest_folder, '{}-z{!s}.mbtiles'.format(ctr, zoom)) |
75 | 72 |
|
76 | 73 | if not op.exists(mbtiles_file_zoomed): |
77 | | - filtered_geo = op.join(dest_folder, '{}.geojson'.format(ctr)) |
| 74 | + filtered_geo = kwargs.get('geojson') or op.join(dest_folder, '{}.geojson'.format(ctr)) |
78 | 75 | fast_parse = [] |
79 | 76 | if not op.exists(filtered_geo): |
80 | 77 | fast_parse = ['-P'] |
@@ -344,7 +341,7 @@ def _tile_results_summary(ml_type, classes): |
344 | 341 |
|
345 | 342 | print('Total tiles: {}'.format(len(all_tiles))) |
346 | 343 |
|
347 | | -def _create_empty_label(ml_type, classes, format): |
| 344 | +def _create_empty_label(ml_type, classes): |
348 | 345 | if ml_type == 'classification': |
349 | 346 | return np.zeros(len(classes) + 1, dtype=np.int) |
350 | 347 | elif ml_type == 'object-detection': |
|
0 commit comments