EiI ^SSKJr SSKJrJrJrJr SSKJrJ r SSK J r J r "SS\ 5r g) )Path)AnyCallableOptionalUnion)default_loader ImageFolder)download_and_extract_archiveverify_str_argc^\rSrSrSrSrSrSSSS\4S\\ \ 4S \ S \ \ S \ \ S \ S \ \ /\4SS4U4SjjjrS\ 4SjrSSjrSrU=r$) 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 or torch.Tensor, depends on the given loader, 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. loader (callable, optional): A function to load an image given its path. By default, it uses PIL as its image loader, but users could also pass in ``torchvision.io.decode_image`` for decoding image data into tensors directly. z;https://openaipublic.azureedge.net/clip/data/country211.tgz 84988d7644798601126c29e9877aab6atrainNFrootsplit transformtarget_transformdownloadloaderreturnc~>[USS5Ul[U5R5n[ U5UlUS- UlU(aUR5 UR5(d [S5e[TU]-[ UR UR- 5UUUS9 [ U5Ulg)Nr)rvalidtest country211z;Dataset not found. You can use download=True to download it)rrr) r _splitr expanduserstrr _base_folder _download _check_exists RuntimeErrorsuper__init__)selfrrrrrr __class__s \/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/datasets/country211.pyr%Country211.__init__ s%UG5OP Dz$$&I  </  NN !!##\] ]  !!DKK/ 0-  I cxURR5=(a URR5$)N)r existsis_dirr&s r(r"Country211._check_exists=s+  '')Hd.?.?.F.F.HHr*cUR5(ag[URURURS9 g)N) download_rootmd5)r"r _URLr_MD5r.s r(r!Country211._download@s,      $TYYdiiTYYWr*)r rr)rN)__name__ __module__ __qualname____firstlineno____doc__r3r4r rrrrrboolrr%r"r!__static_attributes__ __classcell__)r's@r(rrs( ID -D (,/3'5CIH%  #8,   #$ :ItIXXr*rN)pathlibrtypingrrrrfolderr r utilsr r rr*r(rCs"11/?;X;Xr*