Skip to content

Commit

Permalink
chore: Remove obsolete arguments in Pad transformation
Browse files Browse the repository at this point in the history
  • Loading branch information
pierluigiferrari committed Apr 15, 2018
1 parent 9f70e79 commit 8df0198
Showing 1 changed file with 7 additions and 11 deletions.
18 changes: 7 additions & 11 deletions transformations/object_detection_2d_patch_sampling_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -219,7 +219,7 @@ def __init__(self,
patch_xmin,
patch_height,
patch_width,
clip_boxes=False,
clip_boxes=True,
box_filter=None,
background=(0,0,0),
labels_format={'class_id': 0, 'xmin': 1, 'ymin': 2, 'xmax': 3, 'ymax': 4}):
Expand Down Expand Up @@ -360,7 +360,7 @@ def __init__(self,
crop_bottom,
crop_left,
crop_right,
clip_boxes=False,
clip_boxes=True,
box_filter=None,
labels_format={'class_id': 0, 'xmin': 1, 'ymin': 2, 'xmax': 3, 'ymax': 4}):
self.crop_top = crop_top
Expand Down Expand Up @@ -400,24 +400,20 @@ def __init__(self,
pad_bottom,
pad_left,
pad_right,
clip_boxes=False,
box_filter=None,
background=(0,0,0),
labels_format={'class_id': 0, 'xmin': 1, 'ymin': 2, 'xmax': 3, 'ymax': 4}):
self.pad_top = pad_top
self.pad_bottom = pad_bottom
self.pad_left = pad_left
self.pad_right = pad_right
self.clip_boxes = clip_boxes
self.box_filter = box_filter
self.background = background
self.labels_format = labels_format
self.pad = CropPad(patch_ymin=-self.pad_top,
patch_xmin=-self.pad_left,
patch_height=None,
patch_width=None,
clip_boxes=self.clip_boxes,
box_filter=self.box_filter,
clip_boxes=False,
box_filter=None,
background=self.background,
labels_format=self.labels_format)

Expand Down Expand Up @@ -453,7 +449,7 @@ def __init__(self,
box_filter=None,
image_validator=None,
n_trials_max=3,
clip_boxes=False,
clip_boxes=True,
prob=1.0,
background=(0,0,0),
can_fail=False,
Expand Down Expand Up @@ -615,7 +611,7 @@ def __init__(self,
image_validator=None,
bound_generator=None,
n_trials_max=50,
clip_boxes=False,
clip_boxes=True,
prob=0.857,
background=(0,0,0),
labels_format={'class_id': 0, 'xmin': 1, 'ymin': 2, 'xmax': 3, 'ymax': 4}):
Expand Down Expand Up @@ -760,7 +756,7 @@ def __init__(self,
box_filter=None,
image_validator=None,
n_trials_max=3,
clip_boxes=False,
clip_boxes=True,
labels_format={'class_id': 0, 'xmin': 1, 'ymin': 2, 'xmax': 3, 'ymax': 4}):
'''
Arguments:
Expand Down

0 comments on commit 8df0198

Please sign in to comment.