"tests/csrc/vscode:/vscode.git/clone" did not exist on "3253240a8ac3840e001017b5a7b9c5fda371f575"
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
......@@ -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,
......@@ -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)
......
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