Skip to content

GitLab

Unverified Commit 995f9b95 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix test_detection_preset for ssd data augmentation (#7447)

parent 18a2e8eb
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/test_transforms_v2.py
View file @ 995f9b95
...@@ -1875,7 +1875,7 @@ def test_detection_preset(image_type, data_augmentation, to_tensor, sanitize): ...@@ -1875,7 +1875,7 @@ def test_detection_preset(image_type, data_augmentation, to_tensor, sanitize):
elif data_augmentation == "ssd": elif data_augmentation == "ssd":
t = [ t = [
transforms.RandomPhotometricDistort(p=1), 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.RandomIoUCrop(),
transforms.RandomHorizontalFlip(p=1), transforms.RandomHorizontalFlip(p=1),
to_tensor, to_tensor,
...@@ -1934,7 +1934,7 @@ def test_detection_preset(image_type, data_augmentation, to_tensor, sanitize): ...@@ -1934,7 +1934,7 @@ def test_detection_preset(image_type, data_augmentation, to_tensor, sanitize):
# param is True. # param is True.
# Note that the values below are probably specific to the random seed # Note that the values below are probably specific to the random seed
# set above (which is fine). # set above (which is fine).
(True, "ssd"): 4, (True, "ssd"): 5,
(True, "ssdlite"): 4, (True, "ssdlite"): 4,
}.get((sanitize, data_augmentation), num_boxes) }.get((sanitize, data_augmentation), num_boxes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment