New evaluation of descriptors

lzb863 · Nov 15, 2018 · 2be376b · 2be376b
1 parent f20332c
commit 2be376b
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 29 deletions.
diff --git a/notebooks/descriptors_evaluation_on_hpatches.ipynb b/notebooks/descriptors_evaluation_on_hpatches.ipynb
diff --git a/superpoint/configs/superpoint_coco.yaml b/superpoint/configs/superpoint_coco.yaml
@@ -2,7 +2,7 @@ data:
     name: 'coco'
     cache_in_memory: true
     validation_size: 96
-    labels: outputs/mp_synth-v10_ha2-100  # outputs/mp_synth-v6_photo-hom-aug_ha2-100-3-old_coco # outputs/mp_synth-v7_ha1-100_trained_no-topk_coco
+    labels: outputs/mp_synth-v10_ha2-100  # use your own labels here
     augmentation:
         photometric:
             enable: true
@@ -31,15 +31,15 @@ data:
             perspective_amplitude_x: 0.2
             perspective_amplitude_y: 0.2
             patch_ratio: 0.85
-            max_angle: 0.52  # 1.57
+            max_angle: 0.78  # 1.57
             allow_artifacts: true
         valid_border_margin: 3
 model:
     name: 'super_point'
     batch_size: 3
     eval_batch_size: 3
     learning_rate: 0.0001  # 0.001
-    lambda_d: 250
+    lambda_d: 800
     positive_margin: 1
     negative_margin: 0.2
     lambda_loss: 0.0001

diff --git a/superpoint/configs/superpoint_hpatches.yaml b/superpoint/configs/superpoint_hpatches.yaml
@@ -1,7 +1,7 @@
 data:
     name: 'patches_dataset'
     dataset: 'hpatches'  # 'coco' 'hpatches'
-    alteration: 'v'  # 'all' 'i' 'v'
+    alteration: 'i'  # 'all' 'i' 'v'
     preprocessing:
         resize: [480, 640]  # False for coco
 model:
@@ -28,5 +28,5 @@ model:
             patch_ratio: 0.85
             max_angle: 1.57
             allow_artifacts: false
-eval_iter: 50  # 600
+eval_iter: 600
 seed: 1