o
    Wbf                     @   s6   d dl Z ddlmZ d
ddZd
ddZd
dd	ZdS )    N   )_make_swin_backbonec                 C   .   t jd| d}|d krg dn|}t||dS )N*swinv2_large_window12to24_192to384_22kft1k
pretrainedr   r      r   hookstimmcreate_modelr   r   r   model r   UC:\Users\Kristjan\AppData\Local\Programs\Python\MiDaS-master\midas\backbones\swin2.py_make_pretrained_swin2l24_384      r   c                 C   r   )N)swinv2_base_window12to24_192to384_22kft1kr   r   r
   r   r   r   r   r   _make_pretrained_swin2b24_384   r   r   c                 C   s4   t jd| d}|d krg dn|}t||ddgdS )Nswinv2_tiny_window16_256r   )r   r      r   @   )r   
patch_gridr   r   r   r   r   _make_pretrained_swin2t16_256   s   r   )N)r   swin_commonr   r   r   r   r   r   r   r   <module>   s
    



