fix: data aug order to prevent strange bug
Browse files- detector/data.py +1 -1
detector/data.py
CHANGED
@@ -224,10 +224,10 @@ class FontDataset(Dataset):
|
|
224 |
RandomRotate(preserve=0.2),
|
225 |
]
|
226 |
image_transforms = [
|
227 |
-
RandomDownSample(max_ratio=2, preserve=0.5),
|
228 |
torchvision.transforms.GaussianBlur(
|
229 |
random.randint(1, 3) * 2 - 1, sigma=(0.1, 5.0)
|
230 |
),
|
|
|
231 |
torchvision.transforms.Resize((config.INPUT_SIZE, config.INPUT_SIZE)),
|
232 |
torchvision.transforms.ToTensor(),
|
233 |
RandomNoise(max_noise=0.05, preserve=0.1),
|
|
|
224 |
RandomRotate(preserve=0.2),
|
225 |
]
|
226 |
image_transforms = [
|
|
|
227 |
torchvision.transforms.GaussianBlur(
|
228 |
random.randint(1, 3) * 2 - 1, sigma=(0.1, 5.0)
|
229 |
),
|
230 |
+
RandomDownSample(max_ratio=2, preserve=0.5),
|
231 |
torchvision.transforms.Resize((config.INPUT_SIZE, config.INPUT_SIZE)),
|
232 |
torchvision.transforms.ToTensor(),
|
233 |
RandomNoise(max_noise=0.05, preserve=0.1),
|