o
    1ÎjR  ã                   @   sŠ  d dl Z d dlmZ ddlmZmZ ddlmZ ddl	m
Z
 ddlmZ e j d¡Ze e e e¡d	d
¡¡Zede
eeedœƒ edde ed¡g d¢dœƒ edde ed¡g d¢dœƒ edde ed¡g d¢g d¢dddgdœg d¢g dœdg d¢idg d¢ig dœ	ƒ edde ed¡g d¢g d¢dddgdœg d¢g dœdg d¢idg d¢ig dœ	ƒ ed de ed!¡g d¢dœƒ ed"de ed#¡g d¢dœƒ ed$de ed%¡g d¢dœƒ ed&de ed'¡g d¢dœƒ ed(de ed)¡g d¢dœƒ ed*de ed+¡g d¢dœƒ ed,de ed-¡g d¢dœƒ ed.de ed/¡g d¢dœƒ ed0de ed1¡g d¢dœƒ ed2de ed3¡g d¢dœƒ ed4de ed5¡g d¢dœƒ ed6de ed7¡g d¢dœƒ ed8de ed9¡g d¢dœƒ ed:de ed;¡g d¢dœƒ ed<de ed=¡g d¢dœƒ ed>de ed?¡g d¢dœƒ ed@de edA¡g d¢dœƒ dBdC„ ZeddDƒ eddEƒ eddFƒ dS )Gé    Né   )Úregister_model_infoÚregister_suite_infoé   )Ú	SegConfig)ÚSegModel)Ú	SegRunnerZPADDLE_PDX_PADDLESEG_PATHz..ZconfigsZSeg)Z
suite_nameÚmodelÚrunnerÚconfigZrunner_root_pathzOCRNet_HRNet-W48zOCRNet_HRNet-W48.yaml)ÚtrainÚevaluateZpredictZexport)Ú
model_nameÚsuiteÚconfig_pathÚsupported_apiszOCRNet_HRNet-W18zOCRNet_HRNet-W18.yamlzPP-LiteSeg-TzPP-LiteSeg-T.yaml)ÚcpuZgpu_nxcxÚxpuÚnpuÚmluTZO1ZO2)ÚdeviceZdy2stÚamp)r   r   r   )r   Zgpur   r   r   )	r   r   r   r   Zsupported_train_optsZsupported_evaluate_optsZsupported_predict_optsZsupported_infer_optsZsupported_dataset_typeszPP-LiteSeg-BzPP-LiteSeg-B.yamlZSeaFormer_basezSeaFormer_base.yamlZSeaFormer_tinyzSeaFormer_tiny.yamlZSeaFormer_smallzSeaFormer_small.yamlZSeaFormer_largezSeaFormer_large.yamlzSegFormer-B0zSegFormer-B0.yamlzSegFormer-B1zSegFormer-B1.yamlzSegFormer-B2zSegFormer-B2.yamlzSegFormer-B3zSegFormer-B3.yamlzSegFormer-B4zSegFormer-B4.yamlzSegFormer-B5zSegFormer-B5.yamlzDeeplabv3-R50zDeeplabv3-R50.yamlzDeeplabv3-R101zDeeplabv3-R101.yamlzDeeplabv3_Plus-R50zDeeplabv3_Plus-R50.yamlzDeeplabv3_Plus-R101zDeeplabv3_Plus-R101.yamlZSTFPMz
STFPM.yamlZMaskFormer_tinyzMaskFormer_tiny.yamlZMaskFormer_smallzMaskFormer_small.yamlc                 C   s6   ddl m} || ƒ}tdi |¤Ž}||d< t|ƒ d S )Nr   )Úget_registered_model_infor   © )Úbase.registerr   Údictr   )r   Úaliasr   Úrecordr   r   úm/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/repo_apis/PaddleSeg_api/seg/register.pyÚ
_set_alias  s
   r   Zocrnet_hrnetw48Zocrnet_hrnetw18Zpp_liteseg_stdc1)ÚosZos.pathÚpathZospr   r   r   r   r   r	   r   r
   r   ÚenvironÚgetZREPO_ROOT_PATHÚabspathÚjoinÚdirnameÚ__file__ZPDX_CONFIG_DIRr   r   r   r   r   Ú<module>   sv  ûÿ
üÿ	
üÿ

ýþ

ðÿ
ýþ

ðÿ
üÿ	
üÿ	
üÿ	
üÿ

üÿ	
üÿ	
üÿ	
üÿ	
üÿ	
üÿ

üÿ	
üÿ	
üÿ	
üÿ

üÿ

üÿ

üÿ
	
