Skip to content

Commit

Permalink
[fbsync] fix test_detection_preset for ssd data augmentation (#7447)
Browse files Browse the repository at this point in the history
Reviewed By: vmoens

Differential Revision: D44416577

fbshipit-source-id: 3951020abaf72e0634fc96159bd860c0a17aa8e9
  • Loading branch information
NicolasHug authored and facebook-github-bot committed Mar 30, 2023
1 parent 14fad01 commit c4364b1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/test_transforms_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -1875,7 +1875,7 @@ def test_detection_preset(image_type, data_augmentation, to_tensor, sanitize):
elif data_augmentation == "ssd":
t = [
transforms.RandomPhotometricDistort(p=1),
transforms.RandomZoomOut(fill=defaultdict(lambda: (123.0, 117.0, 104.0), {datapoints.Mask: 0})),
transforms.RandomZoomOut(fill=defaultdict(lambda: (123.0, 117.0, 104.0), {datapoints.Mask: 0}), p=1),
transforms.RandomIoUCrop(),
transforms.RandomHorizontalFlip(p=1),
to_tensor,
Expand Down Expand Up @@ -1934,7 +1934,7 @@ def test_detection_preset(image_type, data_augmentation, to_tensor, sanitize):
# param is True.
# Note that the values below are probably specific to the random seed
# set above (which is fine).
(True, "ssd"): 4,
(True, "ssd"): 5,
(True, "ssdlite"): 4,
}.get((sanitize, data_augmentation), num_boxes)

Expand Down

0 comments on commit c4364b1

Please sign in to comment.