Skip to content

Commit a48bd0c

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] Type fix in transformers.py (#6376)
Summary: Update `RandomPhotometricDistort` `__init__` argument to correct types. Reviewed By: datumbox Differential Revision: D38824243 fbshipit-source-id: b0151725296085eb0d6587253fd00fa94084454b
1 parent 6624ed5 commit a48bd0c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

references/detection/transforms.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -218,10 +218,10 @@ def forward(
218218
class RandomPhotometricDistort(nn.Module):
219219
def __init__(
220220
self,
221-
contrast: Tuple[float] = (0.5, 1.5),
222-
saturation: Tuple[float] = (0.5, 1.5),
223-
hue: Tuple[float] = (-0.05, 0.05),
224-
brightness: Tuple[float] = (0.875, 1.125),
221+
contrast: Tuple[float, float] = (0.5, 1.5),
222+
saturation: Tuple[float, float] = (0.5, 1.5),
223+
hue: Tuple[float, float] = (-0.05, 0.05),
224+
brightness: Tuple[float, float] = (0.875, 1.125),
225225
p: float = 0.5,
226226
):
227227
super().__init__()

0 commit comments

Comments
 (0)