o oit @snddlmZddlmZddlmZddlmmZ ddl m Z m Z eGdddZ eGdd d ZdS) ) annotations) dataclass)AnyN)DeviceTensorc@sleZdZUdZded<ded<ded<eddd Zedd d Zedd dZedddZ dddZ dS) DISKFeaturesuA data structure holding DISK keypoints, descriptors and detection scores for an image. Since DISK detects a varying number of keypoints per image, `DISKFeatures` is not batched. Args: keypoints: Tensor of shape :math:`(N, 2)`, where :math:`N` is the number of keypoints. descriptors: Tensor of shape :math:`(N, D)`, where :math:`D` is the descriptor dimension. detection_scores: Tensor of shape :math:`(N,)` where the detection score can be interpreted as the log-probability of keeping a keypoint after it has been proposed (see the paper section *Method → Feature distribution* for details). r keypoints descriptorsdetection_scoresreturnintcCs |jjdS)Nr)rshapeselfrO/home/ubuntu/.local/lib/python3.10/site-packages/kornia/feature/disk/structs.pyn/s zDISKFeatures.nrcCs|jjS)N)rdevicerrrrr3szDISKFeatures.devicecC|jdddfS)zs   7