gyrojeff commited on
Commit
3a87836
·
1 Parent(s): e7affb8

fix: data aug order to prevent strange bug

Browse files
Files changed (1) hide show
  1. 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),