imc2024-playground.ipynb

{"cells":[{"cell_type":"markdown","metadata":{},"source":["# Configuration"]},{"cell_type":"code","execution_count":1,"metadata":{},"outputs":[],"source":["class config:\n","    data=\"./data\"\n","    configs=[\"sift\", \"aliked2k\", \"disk\"]\n","    retrieval=\"dinov2_salad\"\n","    n_retrieval= 30\n","    mode=\"train\"\n","    output=\"./temp\"\n","    pixsfm=True\n","    pixsfm_force=False\n","    pixsfm_max_imgs= 9999\n","    pixsfm_low_mem_threshold=9999\n","    pixsfm_config=\"config/pixsfm.yaml\"\n","    rotation_matching=True\n","    rotation_wrapper=False\n","    resize=None\n","    equalize_transp=True\n","    equalize_grayscale=False\n","    equalize_grid_size=8\n","    equalize_clip_limit=40.\n","    shared_camera=True\n","    overwrite=True\n","    cropping=False\n","    max_rel_crop_size=0.75\n","    min_rel_crop_size=0.2\n","    #skip_scenes=None\n","    skip_scenes=[\"lizard\", \"church\", \"transp_obj_glass_cylinder\", \"transp_obj_glass_cup\", \"multi-temporal-temple-baalshamin\", \"pond\"]\n","    localize_unregistered=True"]},{"cell_type":"markdown","metadata":{},"source":["# Run Pipeline"]},{"cell_type":"code","execution_count":2,"metadata":{"execution":{"iopub.execute_input":"2023-06-12T12:46:50.800157Z","iopub.status.busy":"2023-06-12T12:46:50.799763Z","iopub.status.idle":"2023-06-12T12:46:50.830210Z","shell.execute_reply":"2023-06-12T12:46:50.829370Z","shell.execute_reply.started":"2023-06-12T12:46:50.800117Z"},"trusted":true},"outputs":[],"source":["# Adjust pixsfm config\n","import yaml\n","from pathlib import Path\n","import pixsfm.configs\n","\n","conf_path = Path(pixsfm.configs.__file__).parent / \"low_memory.yaml\"\n","\n","with open(conf_path, 'r') as f:\n","    conf = yaml.load(f, Loader=yaml.FullLoader)\n","\n","conf[\"mapping\"][\"BA\"][\"optimizer\"][\"refine_focal_length\"] = True\n","conf[\"mapping\"][\"BA\"][\"optimizer\"][\"refine_extra_params\"] = True\n","conf[\"mapping\"][\"BA\"][\"optimizer\"][\"refine_extrinsics\"] = True\n","\n","out_conf_path = Path(\"config\", \"pixsfm.yaml\")\n","out_conf_path.parent.mkdir(exist_ok=True, parents=True)\n","with open(out_conf_path, 'w') as f:\n","    yaml.dump(conf, f)"]},{"cell_type":"code","execution_count":3,"metadata":{"execution":{"iopub.execute_input":"2023-06-12T12:52:21.885692Z","iopub.status.busy":"2023-06-12T12:52:21.885006Z","iopub.status.idle":"2023-06-12T13:00:44.713100Z","shell.execute_reply":"2023-06-12T13:00:44.711424Z","shell.execute_reply.started":"2023-06-12T12:52:21.885649Z"},"trusted":true},"outputs":[{"name":"stderr","output_type":"stream","text":["[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] Arguments:\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO]   data: ./data\n","[2024/06/29 13:57:10 root INFO]   configs: ['sift', 'aliked2k', 'disk']\n","[2024/06/29 13:57:10 root INFO]   retrieval: dinov2_salad\n","[2024/06/29 13:57:10 root INFO]   n_retrieval: 30\n","[2024/06/29 13:57:10 root INFO]   mode: train\n","[2024/06/29 13:57:10 root INFO]   output: ./temp\n","[2024/06/29 13:57:10 root INFO]   pixsfm: True\n","[2024/06/29 13:57:10 root INFO]   pixsfm_force: False\n","[2024/06/29 13:57:10 root INFO]   pixsfm_max_imgs: 9999\n","[2024/06/29 13:57:10 root INFO]   pixsfm_low_mem_threshold: 9999\n","[2024/06/29 13:57:10 root INFO]   pixsfm_config: config/pixsfm.yaml\n","[2024/06/29 13:57:10 root INFO]   rotation_matching: True\n","[2024/06/29 13:57:10 root INFO]   rotation_wrapper: False\n","[2024/06/29 13:57:10 root INFO]   resize: None\n","[2024/06/29 13:57:10 root INFO]   equalize_transp: True\n","[2024/06/29 13:57:10 root INFO]   equalize_grayscale: False\n","[2024/06/29 13:57:10 root INFO]   equalize_grid_size: 8\n","[2024/06/29 13:57:10 root INFO]   equalize_clip_limit: 40.0\n","[2024/06/29 13:57:10 root INFO]   shared_camera: True\n","[2024/06/29 13:57:10 root INFO]   overwrite: True\n","[2024/06/29 13:57:10 root INFO]   cropping: False\n","[2024/06/29 13:57:10 root INFO]   max_rel_crop_size: 0.75\n","[2024/06/29 13:57:10 root INFO]   min_rel_crop_size: 0.2\n","[2024/06/29 13:57:10 root INFO]   skip_scenes: ['lizard', 'church', 'transp_obj_glass_cylinder', 'transp_obj_glass_cup', 'multi-temporal-temple-baalshamin', 'pond']\n","[2024/06/29 13:57:10 root INFO]   localize_unregistered: True\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] DATA:\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] transp_obj_glass_cup\n","[2024/06/29 13:57:10 root INFO]   transp_obj_glass_cup: 36 imgs\n","[2024/06/29 13:57:10 root INFO] dioscuri\n","[2024/06/29 13:57:10 root INFO]   dioscuri: 70 imgs\n","[2024/06/29 13:57:10 root INFO] church\n","[2024/06/29 13:57:10 root INFO]   church: 111 imgs\n","[2024/06/29 13:57:10 root INFO] lizard\n","[2024/06/29 13:57:10 root INFO]   lizard: 727 imgs\n","[2024/06/29 13:57:10 root INFO] transp_obj_glass_cylinder\n","[2024/06/29 13:57:10 root INFO]   transp_obj_glass_cylinder: 36 imgs\n","[2024/06/29 13:57:10 root INFO] pond\n","[2024/06/29 13:57:10 root INFO]   pond: 1148 imgs\n","[2024/06/29 13:57:10 root INFO] multi-temporal-temple-baalshamin\n","[2024/06/29 13:57:10 root INFO]   multi-temporal-temple-baalshamin: 75 imgs\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] transp_obj_glass_cup\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] transp_obj_glass_cup - transp_obj_glass_cup\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] Skipping transp_obj_glass_cup - transp_obj_glass_cup\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] dioscuri\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] dioscuri - dioscuri\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] Data paths:\n","[2024/06/29 13:57:10 root INFO]   input_dir: data/train/dioscuri\n","[2024/06/29 13:57:10 root INFO]   scene_dir: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri\n","[2024/06/29 13:57:10 root INFO]   image_dir: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/images\n","[2024/06/29 13:57:10 root INFO]   dataset: dioscuri\n","[2024/06/29 13:57:10 root INFO]   scene: dioscuri\n","[2024/06/29 13:57:10 root INFO]   reference_model: None\n","[2024/06/29 13:57:10 root INFO]   sfm_dir: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/sparse\n","[2024/06/29 13:57:10 root INFO]   pairs_path: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/pairs.txt\n","[2024/06/29 13:57:10 root INFO]   features_retrieval: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/features_retrieval.h5\n","[2024/06/29 13:57:10 root INFO]   features_path: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/features.h5\n","[2024/06/29 13:57:10 root INFO]   matches_path: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/matches.h5\n","[2024/06/29 13:57:10 root INFO]   rotated_image_dir: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/images_rotated\n","[2024/06/29 13:57:10 root INFO]   rotated_features_path: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/features_rotated.h5\n","[2024/06/29 13:57:10 root INFO]   cropping_dir: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/cropping\n","[2024/06/29 13:57:10 root INFO]   cropping_pairs_path: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/cropping/pairs.txt\n","[2024/06/29 13:57:10 root INFO]   cropping_features_retrieval: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/cropping/features_retrieval.h5\n","[2024/06/29 13:57:10 root INFO]   cropping_features_path: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/cropping/features.h5\n","[2024/06/29 13:57:10 root INFO]   cropping_matches_path: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/cropping/matches.h5\n","[2024/06/29 13:57:10 root INFO]   cache: temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/cache\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","[2024/06/29 13:57:10 root INFO] Preprocessing\n","[2024/06/29 13:57:10 root INFO] ================================================================================\n","Rescaling dioscuri: 100%|███████████████████████| 70/70 [00:03<00:00, 17.59it/s]\n","[2024/06/29 13:57:14 root INFO] Images have same shapes: False.\n","[2024/06/29 13:57:14 root INFO] ================================================================================\n","[2024/06/29 13:57:14 root INFO] Rotating images\n","[2024/06/29 13:57:14 root INFO] ================================================================================\n","/home/salva/.venv/image-matching/lib/python3.10/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n","  from .autonotebook import tqdm as notebook_tqdm\n","/home/salva/.venv/image-matching/lib/python3.10/site-packages/timm/models/_factory.py:117: UserWarning: Mapping deprecated model name swsl_resnext50_32x4d to current resnext50_32x4d.fb_swsl_ig1b_ft_in1k.\n","  model = create_fn(\n","/home/salva/.venv/image-matching/lib/python3.10/site-packages/torch/hub.py:682: UserWarning: Falling back to the old format < 1.6. This support will be deprecated in favor of default zipfile format introduced in 1.6. Please redo torch.save() to save it in the new zipfile format.\n","  warnings.warn('Falling back to the old format < 1.6. This support will be '\n","Rotating dioscuri:   0%|                                 | 0/70 [00:00<?, ?it/s]/home/salva/.venv/image-matching/lib/python3.10/site-packages/torchvision/transforms/functional.py:1603: UserWarning: The default value of the antialias parameter of all the resizing transforms (Resize(), RandomResizedCrop(), etc.) will change from None to True in v0.17, in order to be consistent across the PIL and Tensor backends. To suppress this warning, directly pass antialias=True (recommended, future default), antialias=None (current default, which means False for Tensors and True for PIL), or antialias=False (only works on Tensors - PIL will still use antialiasing). This also applies if you are using the inference transforms from the models weights: update the call to weights.transforms(antialias=True).\n","  warnings.warn(\n","Rotating dioscuri: 100%|████████████████████████| 70/70 [00:05<00:00, 11.86it/s]\n","[2024/06/29 13:57:22 root INFO] Rotated 29 of 70 images.\n","[2024/06/29 13:57:22 root INFO] Images have same shapes: False.\n","[2024/06/29 13:57:22 root INFO] ================================================================================\n","[2024/06/29 13:57:22 root INFO] Get pairs\n","[2024/06/29 13:57:22 root INFO] ================================================================================\n","[2024/06/29 13:57:22 hloc INFO] Extracting local features with configuration:\n","{'model': {'name': 'dinov2_salad'},\n"," 'output': 'global-feats-dinov2salad',\n"," 'preprocessing': {'resize_max': 1024}}\n","Using cache found in /home/salva/.cache/torch/hub/serizba_salad_main\n","[2024/06/29 13:57:24 faiss.loader INFO] Loading faiss with AVX2 support.\n","[2024/06/29 13:57:24 faiss.loader INFO] Could not load library with AVX2 support due to:\n","ModuleNotFoundError(\"No module named 'faiss.swigfaiss_avx2'\")\n","[2024/06/29 13:57:24 faiss.loader INFO] Loading faiss.\n","[2024/06/29 13:57:25 faiss.loader INFO] Successfully loaded faiss.\n","Using cache found in /home/salva/.cache/torch/hub/facebookresearch_dinov2_main\n","/home/salva/.cache/torch/hub/facebookresearch_dinov2_main/dinov2/layers/swiglu_ffn.py:43: UserWarning: xFormers is available (SwiGLU)\n","  warnings.warn(\"xFormers is available (SwiGLU)\")\n","/home/salva/.cache/torch/hub/facebookresearch_dinov2_main/dinov2/layers/attention.py:27: UserWarning: xFormers is available (Attention)\n","  warnings.warn(\"xFormers is available (Attention)\")\n","/home/salva/.cache/torch/hub/facebookresearch_dinov2_main/dinov2/layers/block.py:33: UserWarning: xFormers is available (Block)\n","  warnings.warn(\"xFormers is available (Block)\")\n","[2024/06/29 13:57:26 dinov2 INFO] using MLP layer as FFN\n","  0%|          | 0/70 [00:00<?, ?it/s]/home/salva/.venv/image-matching/lib/python3.10/site-packages/torchvision/transforms/functional.py:1603: UserWarning: The default value of the antialias parameter of all the resizing transforms (Resize(), RandomResizedCrop(), etc.) will change from None to True in v0.17, in order to be consistent across the PIL and Tensor backends. To suppress this warning, directly pass antialias=True (recommended, future default), antialias=None (current default, which means False for Tensors and True for PIL), or antialias=False (only works on Tensors - PIL will still use antialiasing). This also applies if you are using the inference transforms from the models weights: update the call to weights.transforms(antialias=True).\n","  warnings.warn(\n","100%|██████████| 70/70 [00:02<00:00, 28.31it/s]\n","[2024/06/29 13:57:31 hloc INFO] Finished exporting features.\n","[2024/06/29 13:57:31 hloc INFO] Extracting image pairs from a retrieval database.\n","[2024/06/29 13:57:31 hloc INFO] Found 2100 pairs.\n","[2024/06/29 13:57:31 root INFO] ================================================================================\n","[2024/06/29 13:57:31 root INFO] Extract features\n","[2024/06/29 13:57:31 root INFO] ================================================================================\n","[2024/06/29 13:57:31 hloc INFO] Extracting local features with configuration:\n","{'model': {'name': 'sift'},\n"," 'options': {'backend': 'pycolmap',\n","             'first_octave': -1,\n","             'peak_threshold': 0.00667},\n"," 'output': 'feats-sift',\n"," 'preprocessing': {'grayscale': True, 'resize_max': 1600}}\n","100%|██████████| 70/70 [00:10<00:00,  6.40it/s]\n","[2024/06/29 13:57:42 hloc INFO] Finished exporting features.\n","[2024/06/29 13:57:42 hloc INFO] Extracting local features with configuration:\n","{'model': {'detection_threshold': 0.0,\n","           'force_num_keypoints': False,\n","           'max_num_keypoints': 2048,\n","           'model_name': 'aliked-n16',\n","           'name': 'aliked'},\n"," 'output': 'feats-aliked2k',\n"," 'preprocessing': {'resize_max': 1600}}\n","100%|██████████| 70/70 [00:03<00:00, 18.48it/s]\n","[2024/06/29 13:57:46 hloc INFO] Finished exporting features.\n","[2024/06/29 13:57:46 hloc INFO] Extracting local features with configuration:\n","{'model': {'max_keypoints': 5000, 'name': 'disk'},\n"," 'output': 'feats-disk',\n"," 'preprocessing': {'grayscale': False, 'resize_max': 1600}}\n","100%|██████████| 70/70 [00:04<00:00, 14.39it/s]\n","[2024/06/29 13:57:51 hloc INFO] Finished exporting features.\n","[2024/06/29 13:57:51 root INFO] ================================================================================\n","[2024/06/29 13:57:51 root INFO] Match features\n","[2024/06/29 13:57:51 root INFO] ================================================================================\n","[2024/06/29 13:57:51 hloc INFO] Matching local features with configuration:\n","{'model': {'depth_confidence': -1,\n","           'features': 'sift',\n","           'filter_threshold': 0.1,\n","           'mp': True,\n","           'name': 'lightglue',\n","           'weights': 'sift_lightglue',\n","           'width_confidence': -1},\n"," 'output': 'matches-sift-lightglue'}\n"]},{"name":"stdout","output_type":"stream","text":["Loaded LightGlue model\n"]},{"name":"stderr","output_type":"stream","text":["100%|██████████| 1322/1322 [01:53<00:00, 11.69it/s]\n","[2024/06/29 13:59:44 hloc INFO] Finished exporting matches.\n","[2024/06/29 13:59:44 hloc INFO] Matching local features with configuration:\n","{'model': {'depth_confidence': -1,\n","           'features': 'aliked',\n","           'filter_threshold': 0.1,\n","           'mp': True,\n","           'name': 'lightglue',\n","           'width_confidence': -1},\n"," 'output': 'matches-aliked2k-lightglue'}\n"]},{"name":"stdout","output_type":"stream","text":["Loaded LightGlue model\n"]},{"name":"stderr","output_type":"stream","text":["100%|██████████| 1322/1322 [01:17<00:00, 17.13it/s]\n","[2024/06/29 14:01:01 hloc INFO] Finished exporting matches.\n","[2024/06/29 14:01:01 hloc INFO] Matching local features with configuration:\n","{'model': {'depth_confidence': -1,\n","           'features': 'disk',\n","           'filter_threshold': 0.1,\n","           'mp': True,\n","           'name': 'lightglue',\n","           'weights': 'disk_lightglue',\n","           'width_confidence': -1},\n"," 'output': 'matches-disk-lightglue'}\n"]},{"name":"stdout","output_type":"stream","text":["Loaded LightGlue model\n"]},{"name":"stderr","output_type":"stream","text":["100%|██████████| 1322/1322 [03:46<00:00,  5.84it/s]\n","[2024/06/29 14:04:48 hloc INFO] Finished exporting matches.\n","[2024/06/29 14:04:48 root INFO] ================================================================================\n","[2024/06/29 14:04:48 root INFO] Creating ensemble\n","[2024/06/29 14:04:48 root INFO] ================================================================================\n","concatenating features: 100%|██████████████████| 70/70 [00:00<00:00, 791.77it/s]\n","[2024/06/29 14:04:50 root INFO] Found 1322 unique pairs\n","[2024/06/29 14:04:50 root INFO] Pairs in matches1: 1322\n","[2024/06/29 14:04:51 root INFO] Pairs in matches2: 1322\n","concatenating matches: 100%|███████████████| 1322/1322 [00:09<00:00, 142.67it/s]\n","concatenating features: 100%|██████████████████| 70/70 [00:00<00:00, 898.22it/s]\n","[2024/06/29 14:05:02 root INFO] Found 1322 unique pairs\n","[2024/06/29 14:05:03 root INFO] Pairs in matches1: 1322\n","[2024/06/29 14:05:03 root INFO] Pairs in matches2: 1322\n","concatenating matches: 100%|███████████████| 1322/1322 [00:08<00:00, 152.55it/s]\n","[2024/06/29 14:05:13 root INFO] ================================================================================\n","[2024/06/29 14:05:13 root INFO] Rotating keypoints\n","[2024/06/29 14:05:13 root INFO] ================================================================================\n","[2024/06/29 14:05:13 root INFO] Using rotated features from temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/features_rotated.h5\n","[2024/06/29 14:05:13 root INFO] Writing rotated keypoints to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/features.h5\n","[2024/06/29 14:05:14 root INFO] ================================================================================\n","[2024/06/29 14:05:14 root INFO] Run SfM\n","[2024/06/29 14:05:14 root INFO] ================================================================================\n","[2024/06/29 14:05:14 root INFO] Not using pixsfm because 29 rotated images are detected\n","[2024/06/29 14:05:14 root INFO] Using images from temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/images\n","[2024/06/29 14:05:14 root INFO] Using pairs from temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/pairs.txt\n","[2024/06/29 14:05:14 root INFO] Using features from temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/features.h5\n","[2024/06/29 14:05:14 root INFO] Using matches from temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/matches.h5\n","[2024/06/29 14:05:14 root INFO] Using CameraMode.AUTO\n","[2024/06/29 14:05:14 hloc INFO] Creating an empty database...\n","[2024/06/29 14:05:14 hloc INFO] Importing images into the database...\n","[2024/06/29 14:05:15 hloc INFO] Importing features into the database...\n","100%|█████████████████████████████████████████| 70/70 [00:00<00:00, 1060.00it/s]\n","[2024/06/29 14:05:15 hloc INFO] Importing matches into the database...\n","100%|█████████████████████████████████████| 1322/1322 [00:01<00:00, 1240.67it/s]\n","[2024/06/29 14:05:17 hloc INFO] Performing geometric verification of the matches...\n","[2024/06/29 14:05:57 hloc INFO] Running 3D reconstruction...\n","[2024/06/29 14:15:04 hloc INFO] Reconstructed 2 model(s).\n","[2024/06/29 14:15:04 hloc INFO] Largest model is #0 with 55 images.\n","[2024/06/29 14:15:04 hloc INFO] Reconstruction statistics:\n","Reconstruction:\n","\tnum_reg_images = 55\n","\tnum_cameras = 55\n","\tnum_points3D = 53115\n","\tnum_observations = 224930\n","\tmean_track_length = 4.23477\n","\tmean_observations_per_image = 4089.64\n","\tmean_reprojection_error = 0.873629\n","\tnum_input_images = 70\n","[2024/06/29 14:15:04 root INFO] ================================================================================\n","[2024/06/29 14:15:04 root INFO] Back-rotate camera poses\n","[2024/06/29 14:15:04 root INFO] ================================================================================\n","[2024/06/29 14:15:04 root INFO] Not using rotation wrapper\n","[2024/06/29 14:15:04 root INFO] ================================================================================\n","[2024/06/29 14:15:04 root INFO] Localize unregistered images\n","[2024/06/29 14:15:04 root INFO] ================================================================================\n","[2024/06/29 14:15:04 root INFO] Found 15 unregistered images\n","[2024/06/29 14:15:04 root INFO] Using HLoc keypoints\n","[2024/06/29 14:15:04 root INFO] Inferring camera for templia__87.png\n","[2024/06/29 14:15:04 root INFO] localizing templia__87.png\n","[2024/06/29 14:15:04 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:04 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:02<00:00,  2.37s/it]\n","[2024/06/29 14:15:06 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:06 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:06 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:06 hloc INFO] Done!\n","[2024/06/29 14:15:06 root INFO] added templia__87.png with 23 inliers, mean inlier dist 0.0954356846473029, 18 db kpts\n","[2024/06/29 14:15:06 root INFO] Inferring camera for templic_30.png\n","[2024/06/29 14:15:06 root INFO] localizing templic_30.png\n","[2024/06/29 14:15:06 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:06 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:02<00:00,  2.67s/it]\n","[2024/06/29 14:15:09 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:09 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:09 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:09 hloc INFO] Done!\n","[2024/06/29 14:15:09 root INFO] added templic_30.png with 21 inliers, mean inlier dist 0.0945945945945946, 17 db kpts\n","[2024/06/29 14:15:09 root INFO] Inferring camera for dioscuria_161.png\n","[2024/06/29 14:15:09 root INFO] localizing dioscuria_161.png\n","[2024/06/29 14:15:09 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:09 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:04<00:00,  4.09s/it]\n","[2024/06/29 14:15:13 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:13 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:13 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:13 hloc INFO] Done!\n","[2024/06/29 14:15:13 root INFO] added dioscuria_161.png with 12 inliers, mean inlier dist 0.061855670103092786, 17 db kpts\n","[2024/06/29 14:15:13 root INFO] Inferring camera for templia__78.png\n","[2024/06/29 14:15:13 root INFO] localizing templia__78.png\n","[2024/06/29 14:15:13 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:13 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:03<00:00,  3.79s/it]\n","[2024/06/29 14:15:17 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:17 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:17 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:17 hloc INFO] Done!\n","[2024/06/29 14:15:17 root INFO] added templia__78.png with 16 inliers, mean inlier dist 0.0730593607305936, 17 db kpts\n","[2024/06/29 14:15:17 root INFO] Inferring camera for templic_23.png\n","[2024/06/29 14:15:17 root INFO] localizing templic_23.png\n","[2024/06/29 14:15:17 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:17 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:00<00:00,  1.03it/s]\n","[2024/06/29 14:15:18 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:18 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:18 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:18 hloc INFO] Done!\n","[2024/06/29 14:15:18 root INFO] added templic_23.png with 14 inliers, mean inlier dist 0.21212121212121213, 17 db kpts\n","[2024/06/29 14:15:18 root INFO] Inferring camera for dioscuria_261.png\n","[2024/06/29 14:15:18 root INFO] localizing dioscuria_261.png\n","[2024/06/29 14:15:18 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:18 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:04<00:00,  4.77s/it]\n","[2024/06/29 14:15:23 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:23 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:23 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:23 hloc INFO] Done!\n","[2024/06/29 14:15:23 root INFO] added dioscuria_261.png with 16 inliers, mean inlier dist 0.06324110671936758, 20 db kpts\n","[2024/06/29 14:15:23 root INFO] Inferring camera for dioscuria_214.png\n","[2024/06/29 14:15:23 root INFO] localizing dioscuria_214.png\n","[2024/06/29 14:15:23 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:23 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:03<00:00,  3.84s/it]\n","[2024/06/29 14:15:27 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:27 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:27 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:27 hloc INFO] Done!\n","[2024/06/29 14:15:27 root INFO] added dioscuria_214.png with 8 inliers, mean inlier dist 0.057971014492753624, 15 db kpts\n","[2024/06/29 14:15:27 root INFO] Inferring camera for templia__75.png\n","[2024/06/29 14:15:27 root INFO] localizing templia__75.png\n","[2024/06/29 14:15:27 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:27 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:03<00:00,  3.66s/it]\n","[2024/06/29 14:15:30 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:30 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:30 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:30 hloc INFO] Done!\n","[2024/06/29 14:15:30 root INFO] added templia__75.png with 18 inliers, mean inlier dist 0.075, 16 db kpts\n","[2024/06/29 14:15:30 root INFO] Inferring camera for dioscuria_147.png\n","[2024/06/29 14:15:30 root INFO] localizing dioscuria_147.png\n","[2024/06/29 14:15:30 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:30 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:02<00:00,  2.45s/it]\n","[2024/06/29 14:15:33 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:33 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:33 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:33 hloc INFO] Done!\n","[2024/06/29 14:15:33 root INFO] added dioscuria_147.png with 20 inliers, mean inlier dist 0.5555555555555556, 1 db kpts\n","[2024/06/29 14:15:33 root INFO] Inferring camera for dioscuria_185.png\n","[2024/06/29 14:15:33 root INFO] localizing dioscuria_185.png\n","[2024/06/29 14:15:33 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:33 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:05<00:00,  5.05s/it]\n","[2024/06/29 14:15:38 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:38 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:38 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:38 hloc INFO] Done!\n","[2024/06/29 14:15:38 root INFO] added dioscuria_185.png with 13 inliers, mean inlier dist 0.047619047619047616, 19 db kpts\n","[2024/06/29 14:15:38 root INFO] Inferring camera for dioscuria_236.png\n","[2024/06/29 14:15:38 root INFO] localizing dioscuria_236.png\n","[2024/06/29 14:15:38 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:38 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:02<00:00,  2.38s/it]\n","[2024/06/29 14:15:40 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:40 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:40 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:40 hloc INFO] Done!\n","[2024/06/29 14:15:40 root INFO] added dioscuria_236.png with 8 inliers, mean inlier dist 0.0898876404494382, 16 db kpts\n","[2024/06/29 14:15:40 root INFO] Inferring camera for templia__94.png\n","[2024/06/29 14:15:40 root INFO] localizing templia__94.png\n","[2024/06/29 14:15:40 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:40 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:03<00:00,  3.04s/it]\n","[2024/06/29 14:15:43 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:43 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:43 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:43 hloc INFO] Done!\n","[2024/06/29 14:15:43 root INFO] added templia__94.png with 18 inliers, mean inlier dist 0.08737864077669903, 18 db kpts\n","[2024/06/29 14:15:43 root INFO] Inferring camera for archive_0054.png\n","[2024/06/29 14:15:43 root INFO] localizing archive_0054.png\n","[2024/06/29 14:15:43 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:43 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:05<00:00,  5.91s/it]\n","[2024/06/29 14:15:49 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:49 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:49 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:49 hloc INFO] Done!\n","[2024/06/29 14:15:49 root INFO] added archive_0054.png with 17 inliers, mean inlier dist 0.046703296703296704, 27 db kpts\n","[2024/06/29 14:15:49 root INFO] Inferring camera for dioscuria_154.png\n","[2024/06/29 14:15:49 root INFO] localizing dioscuria_154.png\n","[2024/06/29 14:15:49 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:49 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:02<00:00,  2.66s/it]\n","[2024/06/29 14:15:52 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:52 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:52 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:52 hloc INFO] Done!\n","[2024/06/29 14:15:52 root INFO] added dioscuria_154.png with 10 inliers, mean inlier dist 0.08333333333333333, 16 db kpts\n","[2024/06/29 14:15:52 root INFO] Inferring camera for dioscuria_228.png\n","[2024/06/29 14:15:52 root INFO] localizing dioscuria_228.png\n","[2024/06/29 14:15:52 hloc INFO] Reading the 3D model...\n","[2024/06/29 14:15:52 hloc INFO] Starting localization...\n","100%|██████████| 1/1 [00:02<00:00,  2.90s/it]\n","[2024/06/29 14:15:55 hloc INFO] Localized 1 / 1 images.\n","[2024/06/29 14:15:55 hloc INFO] Writing poses to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt...\n","[2024/06/29 14:15:55 hloc INFO] Writing logs to temp/aliked2k+disk+sift-rot-pixsfm-sci-loc/dioscuri/dioscuri/loc.txt_logs.pkl...\n","[2024/06/29 14:15:55 hloc INFO] Done!\n","[2024/06/29 14:15:55 root INFO] added dioscuria_228.png with 12 inliers, mean inlier dist 0.08108108108108109, 17 db kpts\n","[2024/06/29 14:15:55 root INFO] Extracting results for dioscuri - dioscuri\n","[2024/06/29 14:15:55 root INFO] Timings:\n","[2024/06/29 14:15:55 root INFO]   preprocessing: 0:00:03.990832 (0.35%)\n","[2024/06/29 14:15:55 root INFO]   crop-images: 0:00:00.000001 (0.00%)\n","[2024/06/29 14:15:55 root INFO]   rotate-images: 0:00:08.249554 (0.73%)\n","[2024/06/29 14:15:55 root INFO]   pairs-extraction: 0:00:08.798705 (0.78%)\n","[2024/06/29 14:15:55 root INFO]   feature-extraction: 0:00:19.687690 (1.75%)\n","[2024/06/29 14:15:55 root INFO]   feature-matching: 0:06:57.538678 (37.11%)\n","[2024/06/29 14:15:55 root INFO]   create-ensemble: 0:00:24.968589 (2.22%)\n","[2024/06/29 14:15:55 root INFO]   rotate-keypoints: 0:00:00.537538 (0.05%)\n","[2024/06/29 14:15:55 root INFO]   sfm: 0:09:50.247202 (52.46%)\n","[2024/06/29 14:15:55 root INFO]   back-rotate-cameras: 0:00:00.003148 (0.00%)\n","[2024/06/29 14:15:55 root INFO]   localize-unreg: 0:00:51.005071 (4.53%)\n","[2024/06/29 14:15:55 root INFO]   Total: 0:18:45.048132\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] church\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] church - church\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] Skipping church - church\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] lizard\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] lizard - lizard\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] Skipping lizard - lizard\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] transp_obj_glass_cylinder\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] transp_obj_glass_cylinder - transp_obj_glass_cylinder\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] Skipping transp_obj_glass_cylinder - transp_obj_glass_cylinder\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] pond\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] pond - pond\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] Skipping pond - pond\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] multi-temporal-temple-baalshamin\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] multi-temporal-temple-baalshamin - multi-temporal-temple-baalshamin\n","[2024/06/29 14:15:55 root INFO] ================================================================================\n","[2024/06/29 14:15:55 root INFO] Skipping multi-temporal-temple-baalshamin - multi-temporal-temple-baalshamin\n","[2024/06/29 14:15:55 root INFO] Written 70 of 2203 images to submission file.\n","[2024/06/29 14:15:55 root INFO] Written 70 of 2203 images to submission file.\n","[2024/06/29 14:15:55 root INFO] transp_obj_glass_cup\n","[2024/06/29 14:15:55 root INFO] dioscuri\n","[2024/06/29 14:15:55 root INFO] \tdioscuri: 70 / 70\n","[2024/06/29 14:15:55 root INFO] Unknown image: lizard->lizard->train/lizard/images/00574.png\n","[2024/06/29 14:15:55 root INFO] Unknown image: lizard->lizard->train/lizard/images/00575.png\n","[2024/06/29 14:15:55 root INFO] Unknown image: lizard->lizard->train/lizard/images/00576.png\n","[2024/06/29 14:15:55 root INFO] Unknown image: lizard->lizard->train/lizard/images/00578.png\n","[2024/06/29 14:15:55 root INFO] Unknown image: lizard->lizard->train/lizard/images/00579.png\n","[2024/06/29 14:15:55 root INFO] Unknown image: lizard->lizard->train/lizard/images/00583.png\n","[2024/06/29 14:15:55 root INFO] Unknown image: lizard->lizard->train/lizard/images/00584.png\n","[2024/06/29 14:15:55 root INFO] *** METRICS ***\n","[2024/06/29 14:15:55 root INFO] transp_obj_glass_cup / transp_obj_glass_cup (36 images, 630 pairs) -> mAA=0.004233, mAA_q=0.004233, mAA_t=0.009524\n","[2024/06/29 14:15:55 root INFO] transp_obj_glass_cup -> mAA_t=0.009524\n","[2024/06/29 14:15:55 root INFO] transp_obj_glass_cylinder / transp_obj_glass_cylinder (36 images, 630 pairs) -> mAA=0.005556, mAA_q=0.005556, mAA_t=0.009524\n","[2024/06/29 14:15:55 root INFO] transp_obj_glass_cylinder -> mAA_t=0.009524\n","[2024/06/29 14:15:56 root INFO] multi-temporal-temple-baalshamin / multi-temporal-temple-baalshamin (66 images, 2145 pairs) -> mAA=0.002875, mAA_q=0.033644, mAA_t=0.003730\n","[2024/06/29 14:15:56 root INFO] multi-temporal-temple-baalshamin -> mAA_t=0.003730\n","[2024/06/29 14:16:21 root INFO] lizard / lizard (704 images, 247456 pairs) -> mAA=0.002737, mAA_q=0.008788, mAA_t=0.028617\n","[2024/06/29 14:16:21 root INFO] lizard -> mAA_t=0.028617\n","[2024/06/29 14:17:26 root INFO] pond / pond (1117 images, 623286 pairs) -> mAA=0.001639, mAA_q=0.006795, mAA_t=0.008712\n","[2024/06/29 14:17:26 root INFO] pond -> mAA_t=0.008712\n","[2024/06/29 14:17:27 root INFO] church / church (110 images, 5995 pairs) -> mAA=0.001168, mAA_q=0.018460, mAA_t=0.003114\n","[2024/06/29 14:17:27 root INFO] church -> mAA_t=0.003114\n","[2024/06/29 14:17:27 root INFO] dioscuri / dioscuri (70 images, 2415 pairs) -> mAA=0.380607, mAA_q=0.625604, mAA_t=0.380883\n","[2024/06/29 14:17:27 root INFO] dioscuri -> mAA_t=0.380883\n","[2024/06/29 14:17:27 root INFO] Final metric -> mAA_t=0.063443 (t: 91.6089129447937 sec.)\n","[2024/06/29 14:17:27 root INFO] Done in 0:20:17.312855\n"]}],"source":["import argparse\n","from pathlib import Path\n","from pixhloc.main import main\n","\n","\n","args = {\n","    \"data\": config.data,\n","    \"configs\": config.configs,\n","    \"retrieval\": config.retrieval,\n","    \"n_retrieval\": config.n_retrieval,\n","    \"mode\": config.mode,\n","    \"output\": config.output,\n","    \"pixsfm\": config.pixsfm,\n","    \"pixsfm_force\": config.pixsfm_force,\n","    \"pixsfm_max_imgs\": config.pixsfm_max_imgs,\n","    \"pixsfm_low_mem_threshold\": config.pixsfm_low_mem_threshold,\n","    \"pixsfm_config\": config.pixsfm_config,\n","    \"rotation_matching\": config.rotation_matching,\n","    \"rotation_wrapper\": config.rotation_wrapper,\n","    \"resize\": config.resize,\n","    \"equalize_transp\": config.equalize_transp,\n","    \"equalize_grayscale\": config.equalize_grayscale,\n","    \"equalize_grid_size\": config.equalize_grid_size,\n","    \"equalize_clip_limit\": config.equalize_clip_limit,\n","    \"shared_camera\": config.shared_camera,\n","    \"overwrite\": config.overwrite,\n","    \"cropping\": config.cropping,\n","    \"max_rel_crop_size\": config.max_rel_crop_size,\n","    \"min_rel_crop_size\": config.min_rel_crop_size,\n","    \"skip_scenes\": config.skip_scenes,\n","    \"localize_unregistered\": config.localize_unregistered\n","}\n","args = argparse.Namespace(**args)\n","Path(args.output).mkdir(exist_ok=True)\n","\n","main(args)"]},{"cell_type":"code","execution_count":4,"metadata":{},"outputs":[{"name":"stderr","output_type":"stream","text":["[2024/06/29 14:17:27 root INFO] Unknown image: lizard->lizard->train/lizard/images/00574.png\n","[2024/06/29 14:17:27 root INFO] Unknown image: lizard->lizard->train/lizard/images/00575.png\n","[2024/06/29 14:17:27 root INFO] Unknown image: lizard->lizard->train/lizard/images/00576.png\n","[2024/06/29 14:17:27 root INFO] Unknown image: lizard->lizard->train/lizard/images/00578.png\n","[2024/06/29 14:17:27 root INFO] Unknown image: lizard->lizard->train/lizard/images/00579.png\n","[2024/06/29 14:17:27 root INFO] Unknown image: lizard->lizard->train/lizard/images/00583.png\n","[2024/06/29 14:17:27 root INFO] Unknown image: lizard->lizard->train/lizard/images/00584.png\n","[2024/06/29 14:17:27 root INFO] *** METRICS ***\n","[2024/06/29 14:17:27 root INFO] transp_obj_glass_cup / transp_obj_glass_cup (36 images, 630 pairs) -> mAA=0.004233, mAA_q=0.004233, mAA_t=0.009524\n","[2024/06/29 14:17:27 root INFO] transp_obj_glass_cup -> mAA_t=0.009524\n","[2024/06/29 14:17:27 root INFO] transp_obj_glass_cylinder / transp_obj_glass_cylinder (36 images, 630 pairs) -> mAA=0.005556, mAA_q=0.005556, mAA_t=0.009524\n","[2024/06/29 14:17:27 root INFO] transp_obj_glass_cylinder -> mAA_t=0.009524\n","[2024/06/29 14:17:27 root INFO] multi-temporal-temple-baalshamin / multi-temporal-temple-baalshamin (66 images, 2145 pairs) -> mAA=0.002875, mAA_q=0.033644, mAA_t=0.003730\n","[2024/06/29 14:17:27 root INFO] multi-temporal-temple-baalshamin -> mAA_t=0.003730\n","[2024/06/29 14:17:53 root INFO] lizard / lizard (704 images, 247456 pairs) -> mAA=0.002737, mAA_q=0.008788, mAA_t=0.028617\n","[2024/06/29 14:17:53 root INFO] lizard -> mAA_t=0.028617\n","[2024/06/29 14:19:03 root INFO] pond / pond (1117 images, 623286 pairs) -> mAA=0.001639, mAA_q=0.006795, mAA_t=0.008712\n","[2024/06/29 14:19:03 root INFO] pond -> mAA_t=0.008712\n","[2024/06/29 14:19:04 root INFO] church / church (110 images, 5995 pairs) -> mAA=0.001168, mAA_q=0.018460, mAA_t=0.003114\n","[2024/06/29 14:19:04 root INFO] church -> mAA_t=0.003114\n","[2024/06/29 14:19:04 root INFO] dioscuri / dioscuri (70 images, 2415 pairs) -> mAA=0.380607, mAA_q=0.625604, mAA_t=0.380883\n","[2024/06/29 14:19:04 root INFO] dioscuri -> mAA_t=0.380883\n","[2024/06/29 14:19:04 root INFO] Final metric -> mAA_t=0.063443 (t: 97.15364408493042 sec.)\n"]},{"data":{"text/plain":["0.0634433"]},"execution_count":4,"metadata":{},"output_type":"execute_result"}],"source":["from pixhloc.utils.eval import eval\n","from pixhloc.utils.utils import setup_logger\n","\n","setup_logger()\n","# EVALUATE\n","eval(submission_csv=\"submission.csv\", data_dir=config.data, verbose=True)"]},{"cell_type":"code","execution_count":null,"metadata":{},"outputs":[],"source":[]}],"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"codemirror_mode":{"name":"ipython","version":3},"file_extension":".py","mimetype":"text/x-python","name":"python","nbconvert_exporter":"python","pygments_lexer":"ipython3","version":"3.10.12"}},"nbformat":4,"nbformat_minor":4}