update gif

Author: LuoXubo

README.md

@@ -61,10 +61,18 @@ See `demo.ipynb` for a quick start.
 - Test video size: 1280x720, 30fps, shot by iPhone 13
 - Test video length: 12s
 
+XFeat + mnn
+
 ![video](assets/xfeat+mnn.gif)
 
+SuperPoint + LightGlue
+
 ![video](assets/sp+lg.gif)
 
+Efficient LoFTR
+
+![video](assets/eloftr.gif)
+
 **Usage:**
 
 ```bash

assets/eloftr.gif

@@ -1,9 +1,22 @@
 torch>=1.11.0
 torchvision>=0.12
 numpy
-opencv-python
+opencv-python==4.10.0.84
+opencv-contrib-python 
 matplotlib
 kornia>=0.6.12
 opencv-contrib-python-headless==4.10.0.84
 tqdm
-opencv-contrib-python 
+albumentations==0.5.1 --no-binary=imgaug,albumentations
+ray>=1.0.1
+einops==0.3.0
+loguru==0.5.3
+yacs>=0.1.8
+autopep8
+pylint
+ipython
+jupyterlab
+h5py==3.1.0
+pytorch-lightning==1.3.5
+torchmetrics==0.6.0  # version problem: https://github.com/NVIDIA/DeepLearningExamples/issues/1113#issuecomment-1102969461
+joblib>=1.0.1

assets/sp+lg.gif

@@ -123,6 +123,9 @@ def eloftr(model, ref, tgt):
         time_mat: time taken for matching
     """
 
+    ref = cv2.cvtColor(ref, cv2.COLOR_RGB2GRAY)
+    tgt = cv2.cvtColor(tgt, cv2.COLOR_RGB2GRAY)
+    
     ref = cv2.resize(ref, (ref.shape[1]//32*32, ref.shape[0]//32*32))
     tgt = cv2.resize(tgt, (tgt.shape[1]//32*32, tgt.shape[0]//32*32))