compvis / kornia /feature /__init__.py
Dexter's picture
Upload folder using huggingface_hub
36c95ba verified
from .affine_shape import LAFAffineShapeEstimator, LAFAffNetShapeEstimator, PatchAffineShapeEstimator
from .defmo import DeFMO
from .hardnet import HardNet, HardNet8
from .integrated import (
get_laf_descriptors,
GFTTAffNetHardNet,
LAFDescriptor,
LocalFeature,
LocalFeatureMatcher,
SIFTFeature,
)
from .laf import (
denormalize_laf,
ellipse_to_laf,
extract_patches_from_pyramid,
extract_patches_simple,
get_laf_center,
get_laf_orientation,
get_laf_scale,
laf_from_center_scale_ori,
laf_from_three_points,
laf_is_inside_image,
laf_to_boundary_points,
laf_to_three_points,
make_upright,
normalize_laf,
perspective_transform_lafs,
raise_error_if_laf_is_not_valid,
scale_laf,
set_laf_orientation,
)
from .loftr import LoFTR
from .matching import DescriptorMatcher, match_mnn, match_nn, match_smnn, match_snn
from .mkd import MKDDescriptor
from .orientation import LAFOrienter, OriNet, PatchDominantGradientOrientation
from .responses import (
BlobDoG,
BlobHessian,
CornerGFTT,
CornerHarris,
dog_response,
gftt_response,
harris_response,
hessian_response,
)
from .scale_space_detector import PassLAF, ScaleSpaceDetector
from .siftdesc import SIFTDescriptor
from .sosnet import SOSNet
from .tfeat import TFeat
__all__ = [
"match_nn",
"match_mnn",
"match_snn",
"match_smnn",
"DescriptorMatcher",
"get_laf_descriptors",
"LAFDescriptor",
"LocalFeature",
"SIFTFeature",
"GFTTAffNetHardNet",
"LocalFeatureMatcher",
"SOSNet",
"harris_response",
"gftt_response",
"hessian_response",
"dog_response",
"CornerHarris",
"CornerGFTT",
"BlobHessian",
"BlobDoG",
"extract_patches_from_pyramid",
"extract_patches_simple",
"normalize_laf",
"denormalize_laf",
"laf_to_boundary_points",
"ellipse_to_laf",
"make_upright",
"get_laf_scale",
"get_laf_center",
"get_laf_orientation",
"set_laf_orientation",
"get_laf_descriptors",
"scale_laf",
"SIFTDescriptor",
"MKDDescriptor",
"HardNet",
"HardNet8",
"DeFMO",
"TFeat",
"OriNet",
"LAFAffNetShapeEstimator",
"PassLAF",
"ScaleSpaceDetector",
"LAFAffineShapeEstimator",
"PatchAffineShapeEstimator",
"LAFOrienter",
"PatchDominantGradientOrientation",
"raise_error_if_laf_is_not_valid",
"laf_is_inside_image",
"laf_from_center_scale_ori",
"laf_to_three_points",
"laf_from_three_points",
"match_nn",
"match_mnn",
"match_snn",
"match_smnn",
"LocalFeatureMatcher",
"LocalFeature",
"SIFTFeature",
"GFTTAffNetHardNet",
"LAFDescriptor",
"DescriptorMatcher",
"LoFTR",
"perspective_transform_lafs",
]