Merge pull request rpautrat#61 from rpautrat/superpoint_v1

Small fix in patch generation
lzb863 · Apr 11, 2019 · c81eab1 · c81eab1
2 parents 17dfc4b + 3e2731a
commit c81eab1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/superpoint/configs/coco_patches_generation.yaml b/superpoint/configs/coco_patches_generation.yaml
@@ -4,7 +4,8 @@ homographies:
     scaling: true
     perspective: true
     scaling_amplitude: 0.1
-    perspective_amplitude: 0.05
+    perspective_amplitude_x: 0.05
+    perspective_amplitude_y: 0.05
     patch_ratio: 0.8
     max_angle: 3.14
 preprocessing:

diff --git a/superpoint/configs/superpoint_coco.yaml b/superpoint/configs/superpoint_coco.yaml
@@ -1,6 +1,6 @@
 data:
     name: 'coco'
-    cache_in_memory: true
+    cache_in_memory: false
     validation_size: 96
     labels: outputs/mp_synth-v10_ha2-100  # use your own labels here
     augmentation:

diff --git a/superpoint/generate_coco_patches.py b/superpoint/generate_coco_patches.py
@@ -6,8 +6,8 @@
 import yaml
 from pathlib import Path
 
-from superpoint.models.utils import (sample_homography, flat2mat,
-                                     invert_homography)
+from superpoint.models.homographies import (sample_homography, flat2mat,
+                                            invert_homography)
 from superpoint.settings import DATA_PATH