Fix indentation error.

chiebot
YuAng 3 years ago
parent 467b3d2107
commit 6d4b63e87b
  1. 1
      .gitignore
  2. 1
      src/loftr/utils/coarse_matching.py
  3. 4
      src/utils/dataset.py

1
.gitignore vendored

@ -24,3 +24,4 @@ tools/draw_QccDayNights.py
scripts/slurm/
scripts/sbatch_submit.sh
src/utils/client.py

@ -197,6 +197,7 @@ class CoarseMatching(nn.Module):
# 4. Random sampling of training samples for fine-level LoFTR
# (optional) pad samples with gt coarse-level matches
if self.training:
# NOTE:
# The sampling is performed across all pairs in a batch without manually balancing
# #samples for fine-level increases w.r.t. batch_size

@ -8,6 +8,10 @@ import torch
from numpy.linalg import inv
try:
# for internel use only
from .client import MEGADEPTH_CLIENT, SCANNET_CLIENT
except Exception:
MEGADEPTH_CLIENT = SCANNET_CLIENT = None
# --- DATA IO ---

Loading…
Cancel
Save