o
    )jo                     @   s~   d dl Z d dlmZmZmZmZ d dlZd dlmZ d dl	m
Z
 d dlmZ d dlmZ ejejejdG dd	 d	e
ZdS )
    N)AnyDictOptionalUnion)Models)
TorchModel)MODELS)Tasks)module_namec                       s*   e Zd Zdef fddZdd Z  ZS )HRNetCrowdCounting	model_dirc                    sf   t  j|fi | ddlm} tj|d}|dd|d}|tj	tj|ddd	d
 || _
d S )N   )HighResolutionNetz$average_clip_domain_center_54.97.npzg      ?r   )Zattn_weightZ
fix_domaindomain_center_modelzDCANet_final.pthcpuT)Zmap_locationZweights_only)super__init__Zhrnet_aspp_relur   ospathjoinZload_state_dicttorchloadmodel)selfr   kwargsZHRNet_aspp_relur   net	__class__ m/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/crowd_counting/cc_model.pyr      s$   
zHRNetCrowdCounting.__init__c                 C   s
   |  |S )N)r   )r   Zinputsr   r   r   forward$   s   
zHRNetCrowdCounting.forward)__name__
__module____qualname__strr   r    __classcell__r   r   r   r   r      s    r   )r   typingr   r   r   r   r   Zmodelscope.metainfor   Z'modelscope.models.base.base_torch_modelr   Zmodelscope.models.builderr   Zmodelscope.utils.constantr	   Zregister_moduleZcrowd_countingr   r   r   r   r   <module>   s   