chore: Make clip_boxes default to True

pierluigiferrari · Apr 15, 2018 · 65aa583 · 65aa583
1 parent cd5de00
commit 65aa583
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/data_augmentation_chains/data_augmentation_chain_constant_input_size.py b/data_augmentation_chains/data_augmentation_chain_constant_input_size.py
@@ -43,7 +43,7 @@ def __init__(self,
                  random_translate=((0.03,0.5), (0.03,0.5), 0.5),
                  random_scale=(0.5, 2.0, 0.5),
                  n_trials_max=3,
-                 clip_boxes=False,
+                 clip_boxes=True,
                  overlap_criterion='area',
                  bounds_box_filter=(0.3, 1.0),
                  bounds_validator=(0.5, 1.0),

diff --git a/data_augmentation_chains/data_augmentation_chain_original_ssd.py b/data_augmentation_chains/data_augmentation_chain_original_ssd.py
@@ -93,7 +93,7 @@ def __init__(self, labels_format={'class_id': 0, 'xmin': 1, 'ymin': 2, 'xmax': 3
                                           image_validator=self.image_validator,
                                           bound_generator=self.bound_generator,
                                           n_trials_max=50,
-                                          clip_boxes=False,
+                                          clip_boxes=True,
                                           prob=0.857,
                                           labels_format=self.labels_format)
 

diff --git a/data_augmentation_chains/data_augmentation_chain_variable_input_size.py b/data_augmentation_chains/data_augmentation_chain_variable_input_size.py
@@ -48,7 +48,7 @@ def __init__(self,
                  min_aspect_ratio = 0.5,
                  max_aspect_ratio = 2.0,
                  n_trials_max=3,
-                 clip_boxes=False,
+                 clip_boxes=True,
                  overlap_criterion='area',
                  bounds_box_filter=(0.3, 1.0),
                  bounds_validator=(0.5, 1.0),