
    i*
                     f    d dl mZ ddlmZ ddlmZ  ed      e G d de                    ZdgZy	)
    )strict   )PreTrainedConfig)auto_docstringzmagic-leap-community/superpoint)
checkpointc                       e Zd ZU dZdZdZee   eedf   z  e	d<   dZ
ee	d<   dZee	d	<   dZee	d
<   dZee	d<   dZee	d<   dZee	d<   dZee	d<   dZee	d<   y)SuperPointConfiga)  
    encoder_hidden_sizes (`List`, *optional*, defaults to `[64, 64, 128, 128]`):
        The number of channels in each convolutional layer in the encoder.
    keypoint_decoder_dim (`int`, *optional*, defaults to 65):
        The output dimension of the keypoint decoder.
    descriptor_decoder_dim (`int`, *optional*, defaults to 256):
        The output dimension of the descriptor decoder.
    keypoint_threshold (`float`, *optional*, defaults to 0.005):
        The threshold to use for extracting keypoints.
    max_keypoints (`int`, *optional*, defaults to -1):
        The maximum number of keypoints to extract. If `-1`, will extract all keypoints.
    nms_radius (`int`, *optional*, defaults to 4):
        The radius for non-maximum suppression.
    border_removal_distance (`int`, *optional*, defaults to 4):
        The distance from the border to remove keypoints.

    Example:
    ```python
    >>> from transformers import SuperPointConfig, SuperPointForKeypointDetection

    >>> # Initializing a SuperPoint superpoint style configuration
    >>> configuration = SuperPointConfig()
    >>> # Initializing a model from the superpoint style configuration
    >>> model = SuperPointForKeypointDetection(configuration)
    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```
superpoint)@   r      r   .encoder_hidden_sizes   decoder_hidden_sizeA   keypoint_decoder_dimdescriptor_decoder_dimg{Gzt?keypoint_thresholdmax_keypoints   
nms_radiusborder_removal_distanceg{Gz?initializer_rangeN)__name__
__module____qualname____doc__
model_typer   listinttuple__annotations__r   r   r   r   floatr   r   r   r        /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/superpoint/configuration_superpoint.pyr	   r	      s    8 J8J$s)eCHo5J"" "#""%C% %%M3J#$S$#u#r%   r	   N)huggingface_hub.dataclassesr   configuration_utilsr   utilsr   r	   __all__r$   r%   r&   <module>r+      sE     / 3 # <='$' '$  >'$T 
r%   