Skip to content

GitLab

Unverified Commit 96dbada4 authored by vcwai's avatar vcwai Committed by GitHub
Browse files

Type fix in transformers.py (#6376) 

Update `RandomPhotometricDistort` `__init__` argument to correct types.
parent 96aa3d92
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
references/detection/transforms.py
View file @ 96dbada4
...@@ -218,10 +218,10 @@ class RandomZoomOut(nn.Module): ...@@ -218,10 +218,10 @@ class RandomZoomOut(nn.Module):
class RandomPhotometricDistort(nn.Module): class RandomPhotometricDistort(nn.Module):
def __init__( def __init__(
self, self,
contrast: Tuple[float] = (0.5, 1.5), contrast: Tuple[float, float] = (0.5, 1.5),
saturation: Tuple[float] = (0.5, 1.5), saturation: Tuple[float, float] = (0.5, 1.5),
hue: Tuple[float] = (-0.05, 0.05), hue: Tuple[float, float] = (-0.05, 0.05),
brightness: Tuple[float] = (0.875, 1.125), brightness: Tuple[float, float] = (0.875, 1.125),
p: float = 0.5, p: float = 0.5,
): ):
super().__init__() super().__init__()
......
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