hloc/__init__.py

import logging

import torch
from packaging import version

__version__ = "1.3"

formatter = logging.Formatter(
    fmt="[%(asctime)s %(name)s %(levelname)s] %(message)s",
    datefmt="%Y/%m/%d %H:%M:%S",
)
handler = logging.StreamHandler()
handler.setFormatter(formatter)
handler.setLevel(logging.INFO)

logger = logging.getLogger("hloc")
logger.setLevel(logging.INFO)
logger.addHandler(handler)
logger.propagate = False

try:
    import pycolmap
except ImportError:
    logger.warning("pycolmap is not installed, some features may not work.")
else:
    minimal_version = version.parse("0.3.0")
    found_version = version.parse(getattr(pycolmap, "__version__"))
    if found_version < minimal_version:
        logger.warning(
            "hloc now requires pycolmap>=%s but found pycolmap==%s, "
            "please upgrade with `pip install --upgrade pycolmap`",
            minimal_version,
            found_version,
        )

DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")