Skip to content

Commit 2d24e2b

Browse files
jzxucopybara-github
authored andcommitted
Add ability to create labeling images with bounding boxes based on building footprints.
PiperOrigin-RevId: 846306254
1 parent 180accd commit 2d24e2b

3 files changed

Lines changed: 189 additions & 112 deletions

File tree

src/create_labeling_examples.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,11 @@
114114
' filter. The column must contain binary values, either true/false or 0/1.'
115115
' Rows with positive values in this column are then filtered out.',
116116
)
117+
flags.DEFINE_bool(
118+
'output_images_with_footprints',
119+
False,
120+
'If true, write a set of labeling images with footprints.',
121+
)
117122

118123

119124
def validate_score_bins(score_bins_to_sample_fraction):
@@ -196,6 +201,7 @@ def main(unused_argv):
196201
score_bins_to_sample_fraction,
197202
FLAGS.scores_path,
198203
FLAGS.filter_by_column,
204+
FLAGS.output_images_with_footprints,
199205
)
200206

201207
if num_images == 0:

0 commit comments

Comments
 (0)