o
    ÓÙ¾iÙ  ã                   @   s4  d dl mZmZmZmZmZmZ d dlmZm	Z	 d dl
T d dlmZmZmZ d dlmZ d dlmZmZ d dlmZmZ d dlmZ d d	lmZmZ d d
lmZmZ d dlm Z  d dl!m"Z"m#Z#m$Z$ d dl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, d dl-m.Z. d dl-m/Z/m0Z0m1Z1m2Z2m3Z3 d dl4m5Z5 d dl6T d dl7m8Z8 dS )é   )ÚRandAugmentÚAutoAugmentÚrand_augment_opsÚauto_augment_policyÚrand_augment_transformÚauto_augment_transform)Úresolve_data_configÚresolve_model_data_config)Ú*)ÚImageDatasetÚIterableImageDatasetÚAugMixDataset)Úcreate_dataset)ÚDatasetInfoÚCustomDatasetInfo)ÚImageNetInfoÚinfer_imagenet_subset)Úcreate_loader)ÚMixupÚFastCollateMixup)ÚNaFlexMapDatasetWrapperÚcalculate_naflex_batch_size)Úcreate_naflex_loader)ÚNaFlexMixupÚpairwise_mixup_targetÚmix_batch_variable_size)ÚResizeToSequenceÚCenterCropToSequenceÚRandomCropToSequenceÚRandomResizedCropToSequenceÚResizeKeepRatioToSequenceÚPatchifyÚpatchify_image)Úcreate_reader)Úget_img_extensionsÚis_img_extensionÚset_img_extensionsÚadd_img_extensionsÚdel_img_extensions)ÚRealLabelsImagenet)Úcreate_transformN)9Úauto_augmentr   r   r   r   r   r   Úconfigr   r	   Ú	constantsÚdatasetr   r   r   Údataset_factoryr   Údataset_infor   r   Úimagenet_infor   r   Úloaderr   Úmixupr   r   Únaflex_datasetr   r   Únaflex_loaderr   Únaflex_mixupr   r   r   Únaflex_transformsr   r   r   r   r    r!   r"   Úreadersr#   r$   r%   r&   r'   r(   Úreal_labelsr)   Ú
transformsÚtransforms_factoryr*   © r<   r<   úF/home/ubuntu/.local/lib/python3.10/site-packages/timm/data/__init__.pyÚ<module>   s$     $	