cvg / LightGlue

LightGlue: Local Feature Matching at Light Speed (ICCV 2023)
Apache License 2.0
3.15k stars 291 forks source link

No keypoints with sift = error #114

Open ExtReMLapin opened 4 months ago

ExtReMLapin commented 4 months ago

Hello, only when using sift I get this error on some images:


Traceback (most recent call last):
  File "server.py", line 510, in post
    scores = classify_from_color_layers(colors, just_do_class=force_class_param)
  File "server.py", line 219, in classify_from_color_layers
    image_features = get_features(image_np)
  File "server.py", line 84, in get_features
    feats0 = extractor.extract(image0.to(device))
  File "venv\lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
  File "venv\lib\site-packages\lightglue\utils.py", line 144, in extract
    feats = self.forward({"image": img})
  File "venv\lib\site-packages\lightglue\sift.py", line 211, in forward
    p = self.extract_single_image(img)
  File "venv\lib\site-packages\lightglue\sift.py", line 178, in extract_single_image
    keep = filter_dog_point(
  File "venv\lib\site-packages\lightglue\sift.py", line 25, in filter_dog_point
    np.maximum.at(buffer, tuple(ij), s)
ValueError: array is not broadcastable to correct shape

Tried to fix it myself but I failed, pred["keypoints"] len is 0