o 0i @sPddlmZddlmZmZmZddlmZddlm Z m Z GdddeZ dS) )Path)CallableOptionalUnion) ImageFolder)download_and_extract_archiveverify_str_argc steZdZdZdZdZ    ddeeefded e e d e e d e d df fd d Z d e fddZ dddZZS) Country211a`The Country211 Data Set `_ from OpenAI. This dataset was built by filtering the images from the YFCC100m dataset that have GPS coordinate corresponding to a ISO-3166 country code. The dataset is balanced by sampling 150 train images, 50 validation images, and 100 test images for each country. Args: root (str or ``pathlib.Path``): Root directory of the dataset. split (string, optional): The dataset split, supports ``"train"`` (default), ``"valid"`` and ``"test"``. transform (callable, optional): A function/transform that takes in a PIL image and returns a transformed version. E.g, ``transforms.RandomCrop``. target_transform (callable, optional): A function/transform that takes in the target and transforms it. download (bool, optional): If True, downloads the dataset from the internet and puts it into ``root/country211/``. If dataset is already downloaded, it is not downloaded again. z;https://openaipublic.azureedge.net/clip/data/country211.tgz 84988d7644798601126c29e9877aab6atrainNFrootsplit transformtarget_transformdownloadreturncsvt|dd|_t|}t||_|d|_|r||s%t dt j t|j|j||dt||_dS)Nr)r validtest country211z;Dataset not found. You can use download=True to download it)rr) r _splitr expanduserstrr _base_folder _download _check_exists RuntimeErrorsuper__init__)selfr rrrr __class__]/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torchvision/datasets/country211.pyrs   zCountry211.__init__cCs|jo |jS)N)rexistsis_dirrr"r"r#r4szCountry211._check_existscCs$|rdSt|j|j|jddS)N) download_rootmd5)rr_URLr _MD5r&r"r"r#r7szCountry211._download)r NNF)rN)__name__ __module__ __qualname____doc__r)r*rrrrrboolrrr __classcell__r"r"r r#r s. r N) pathlibrtypingrrrfolderrutilsrr r r"r"r"r#s