Spaces:
Running
Running
File size: 1,003 Bytes
9223079 8320ccc b075789 9223079 e15a186 9223079 b075789 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
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")
|