fix: display lines

common/utils.py

@@ -20,10 +20,14 @@ from .viz import (
 )
 import time
 import matplotlib.pyplot as plt
+import warnings
+warnings.filterwarnings("ignore", category=UserWarning)
+warnings.filterwarnings("ignore", category=gr.deprecation.GradioDeprecationWarning)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 ROOT = Path(__file__).parent.parent
+# some default values
 DEFAULT_SETTING_THRESHOLD = 0.1
 DEFAULT_SETTING_MAX_FEATURES = 2000
 DEFAULT_DEFAULT_KEYPOINT_THRESHOLD = 0.01

common/viz.py

@@ -1,8 +1,10 @@
+import cv2
+import matplotlib
 import numpy as np
 import seaborn as sns
-import matplotlib
 import matplotlib.pyplot as plt
 from pathlib import Path
+import typing
 from typing import Dict, Any, Optional, Tuple, List, Union