Skip to content

Commit 9266236

Browse files
committed
make tiles subdir and provide default for imagery_offset arg
1 parent 80ce1b0 commit 9266236

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

label_maker/preview.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from label_maker.utils import class_match, download_tile_tms, get_tile_tif, is_tif
1111

12-
def preview(dest_folder, number, classes, imagery, ml_type, imagery_offset, **kwargs):
12+
def preview(dest_folder, number, classes, imagery, ml_type, imagery_offset=False, **kwargs):
1313
"""Produce imagery examples for specified classes
1414
1515
Parameters
@@ -55,8 +55,12 @@ def preview(dest_folder, number, classes, imagery, ml_type, imagery_offset, **kw
5555
for i, cl in enumerate(classes):
5656
# create class directory
5757
class_dir = op.join(dest_folder, 'examples', cl.get('name'))
58-
if not op.isdir(class_dir):
59-
makedirs(class_dir)
58+
59+
# the download process downloads into a tiles subdirectory so create that here
60+
class_tile_dir = op.join(class_dir, 'tiles')
61+
62+
if not op.isdir(class_tile_dir):
63+
makedirs(class_tile_dir)
6064

6165
class_tiles = (t for t in tiles.files
6266
if class_match(ml_type, tiles[t], i + 1))

0 commit comments

Comments
 (0)