o piH @s<ddlZddlZddlmZddlmZGdddeZdS)N) BaseDataset) _urlretrievec@s(eZdZdZddZddZddZdS) Cars196z^https://www.kaggle.com/api/v1/datasets/download/jutrera/stanford-car-dataset-by-classes-folderc Cs |jdkr ttdd}n|jdkrttdd}nttdd}ttj|jdd}dd |D}Wdn1s@wY|j tj|jd |dd \}}|j tj|jd |dd \}}||}||} gg|_ |_ t || D]\} } | |vr|j | |j | qydS) Ntrainctestz names.csvrcSsg|]}|qS)strip).0xr r \/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_metric_learning/datasets/cars196.py sz*Cars196.generate_split..zanno_train.csv)splitz anno_test.csv)rsetrangeopenospathjoinroot readlines _load_csvpathslabelszipappend) selfclassesfnames paths_train labels_train paths_test labels_testrrplr r rgenerate_split s.      zCars196.generate_splitc Csgg}}t|d=}|D]/}|d}|d} |d} |tj|jdd||t| ddd| |t| q Wd||fS1sJwY||fS) Nr ,rcar_datar/-) rrrrrrrintreplace) r rr#r all_paths all_labelsr"r) path_annos curr_path curr_labelr r rr's,     zCars196._load_csvcCstj|jddtj|jtjdd}ttj|dt |d}| |jWdn1s4wYt |dS)NT)exist_okr.r,)urlfilenamer ) rmakedirsrrrr DOWNLOAD_URLrrzipfileZipFile extractallremove)r download_folder_pathzip_refr r rdownload_and_remove;szCars196.download_and_removeN)__name__ __module__ __qualname__r;r*rrBr r r rrs  r)rr<datasets.base_datasetrutils.common_functionsrrr r r rs