Fixing #37

cvg · Nov 1, 2023 · 0d93e4c · 0d93e4c
1 parent 4a82835
commit 0d93e4c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/gluefactory/datasets/augmentations.py b/gluefactory/datasets/augmentations.py
@@ -191,7 +191,13 @@ def _init(self, conf):
  [
  A.Blur(**kwi(blur, p=0.1, blur_limit=(3, 9), n="blur")),
  A.MotionBlur(
- **kwi(blur, p=0.2, blur_limit=(3, 25), n="motion_blur")
+ **kwi(
+ blur,
+ p=0.2,
+ blur_limit=(3, 25),
+ allow_shifted=False,
+ n="motion_blur",
+ )
  ),
  A.ISONoise(),
  A.ImageCompression(),
@@ -222,14 +228,14 @@ def _init(self, conf):
  A.OneOf(
  [
  A.Blur(blur_limit=(3, 9)),
- A.MotionBlur(blur_limit=(3, 25)),
+ A.MotionBlur(blur_limit=(3, 25), allow_shifted=False),
  A.ISONoise(),
  A.ImageCompression(),
  ],
  p=0.1,
  ),
  A.Blur(p=0.1, blur_limit=(3, 9)),
- A.MotionBlur(p=0.1, blur_limit=(3, 25)),
+ A.MotionBlur(p=0.1, blur_limit=(3, 25), allow_shifted=False),
  A.RandomBrightnessContrast(
  p=0.5, brightness_limit=(-0.4, 0.0), contrast_limit=(-0.3, 0.0)
  ),

diff --git a/gluefactory/datasets/homographies.py b/gluefactory/datasets/homographies.py
@@ -293,7 +293,7 @@ def visualize(args):
  images = []
  for _, data in zip(range(args.num_items), loader):
  images.append(
- (data[f"view{i}"]["image"][0].permute(1, 2, 0) for i in range(2))
+ [data[f"view{i}"]["image"][0].permute(1, 2, 0) for i in range(2)]
  )
  plot_image_grid(images, dpi=args.dpi)
  plt.tight_layout()

diff --git a/gluefactory/datasets/hpatches.py b/gluefactory/datasets/hpatches.py
@@ -127,7 +127,7 @@ def visualize(args):
  images = []
  for _, data in zip(range(args.num_items), loader):
  images.append(
- (data[f"view{i}"]["image"][0].permute(1, 2, 0) for i in range(2))
+ [data[f"view{i}"]["image"][0].permute(1, 2, 0) for i in range(2)]
  )
  plot_image_grid(images, dpi=args.dpi)
  plt.tight_layout()