o èp©iWã@s<ddlZddlZddlmZddlmZGdd„deƒZdS)éNé)Ú BaseDataset)Ú _urlretrievec@s eZdZdZdd„Zdd„ZdS)ÚCUBzChttps://data.caltech.edu/records/65de6-vp158/files/CUB_200_2011.tgzc Csbtj d¡d dd¡}|jdkrttddƒƒ}n|jdkr&ttdd ƒƒ}nttdd ƒƒ}gg|_|_tt j   |j |d ¡ƒh}tt j   |j |d ¡ƒD}t ||ƒD]6\}}ttt| ¡ƒƒ\}}||vreqQ| ¡\} } t| ƒ} || kr‡|j t j   |j |d | ¡¡|j |¡qQWdƒn1s’wYWdƒdSWdƒdS1sªwYdS) Nú/éÿÿÿÿz.tgzÚÚtrainééeÚtestéÉzimage_class_labels.txtz images.txtÚimages)rÚ DOWNLOAD_URLÚsplitÚreplaceÚsetÚrangeÚpathsÚlabelsÚopenÚosÚpathÚjoinÚrootÚzipÚlistÚmapÚintÚappend) ÚselfÚdir_nameÚclassesÚf1Úf2Úl1Úl2Úimg_idx1Ú class_idxÚimg_idx2Úimg_path©r+úX/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_metric_learning/datasets/cub.pyÚgenerate_split s4   ÿ €òÿÿ"ÿzCUB.generate_splitcCs€tj|jddtj |jtj d¡d¡}ttj|dt   |d¡}|  |j¡Wdƒn1s4wYt  |¡dS)NT)Úexist_okrr)ÚurlÚfilenamezr:gz) rÚmakedirsrrrrrrrÚtarfilerÚ extractallÚremove)r Údownload_folder_pathÚtarr+r+r,Údownload_and_remove+sÿzCUB.download_and_removeN)Ú__name__Ú __module__Ú __qualname__rr-r7r+r+r+r,rs r)rr2Údatasets.base_datasetrÚutils.common_functionsrrr+r+r+r,Ús