File size: 685 Bytes
9223079 |
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 |
import sys
from pathlib import Path
from ..utils.base_model import BaseModel
sys.path.append(str(Path(__file__).parent / "../../third_party"))
from SuperGluePretrainedNetwork.models.superglue import SuperGlue as SG
class SuperGlue(BaseModel):
default_conf = {
"weights": "outdoor",
"sinkhorn_iterations": 100,
"match_threshold": 0.2,
}
required_inputs = [
"image0",
"keypoints0",
"scores0",
"descriptors0",
"image1",
"keypoints1",
"scores1",
"descriptors1",
]
def _init(self, conf):
self.net = SG(conf)
def _forward(self, data):
return self.net(data)
|