U h"@sddlZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZGdd d eZGd d d eZdS) N)Path)AnyCallableListOptionalTupleUnion)Image)download_and_extract_archiveverify_str_arg) VisionDatasetcseZdZdZdeeefeeeefee ee e ddfdd Z e e eefdd d Ze d d d Ze d ddZdd ddZed ddZZS) Caltech101a?`Caltech 101 `_ Dataset. .. warning:: This class needs `scipy `_ to load target files from `.mat` format. Args: root (str or ``pathlib.Path``): Root directory of dataset where directory ``caltech101`` exists or will be saved to if download is set to True. target_type (string or list, optional): Type of target to use, ``category`` or ``annotation``. Can also be a list to output a tuple with all specified target types. ``category`` represents the target class, and ``annotation`` is a list of points from a hand-generated outline. Defaults to ``category``. 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 in root directory. If dataset is already downloaded, it is not downloaded again. .. warning:: To download the dataset `gdown `_ is required. categoryNF)root target_type transformtarget_transformdownloadreturnc s tjtj|d||dtj|jddt|tr<|g}dd|D|_ |rX| | sht dt ttj|jd|_|jd d d d d dttfdd|j|_g|_g|_t|jD]L\}}tttj|jd|}|jtd|d|j||gqdS)NZ caltech101rrTexist_okcSsg|]}t|ddqS)r)r annotation)r ).0trN/var/www/html/venv/lib/python3.8/site-packages/torchvision/datasets/caltech.py 4sz'Caltech101.__init__..HDataset not found or corrupted. You can use download=True to download it101_ObjectCategoriesZBACKGROUND_GoogleZFaces_2ZFaces_3Z Motorbikes_16ZAirplanes_Side_2)ZFacesZ Faces_easyZ MotorbikesZ airplanescs|kr|S|SNr)xZname_maprrHz%Caltech101.__init__..r )super__init__ospathjoinmakedirsr isinstancestrrr_check_integrity RuntimeErrorsortedlistdir categoriesremovelistmapannotation_categoriesindexy enumeratelenextendrange) selfrrrrricn __class__r#rr'(s.  zCaltech101.__init__r7rc Csddl}ttj|jd|j|j|d|j |dd}g}|j D]l}|dkrf| |j|qH|dkrH|j tj|jd |j|j|d |j |dd }| |d qHt|d krt|n|d}|jdk r||}|jdk r||}||fS)z Args: index (int): Index Returns: tuple: (image, target) where the type of target specified by target_type. rNr Zimage_04d.jpgrrZ AnnotationsZ annotation_z.matZ obj_contourr )Zscipy.ior openr(r)r*rr2r8r7rappendioZloadmatr6r:tuplerr)r=r7Zscipyimgtargetrdatarrr __getitem__Qs:      zCaltech101.__getitem__rcCstjtj|jdS)Nr r(r)existsr*rr=rrrr.|szCaltech101._check_integritycCs t|jSr!r:r7rQrrr__len__szCaltech101.__len__cCs<|rtddStd|jdddtd|jddddS) N%Files already downloaded and verifiedzAhttps://drive.google.com/file/d/137RyRjvTBkBiIfeYBNZBtViDHQ6_Ewspz101_ObjectCategories.tar.gzZ b224c7392d521a49829488ab0f1120d9filenamemd5zAhttps://drive.google.com/file/d/175kQy3UsZ0wUEHZjqkUDdNVssr7bgh_mzAnnotations.tarZ 6f83eeb1f24d99cab4eb377263132c91r.printr rrQrrrrszCaltech101.downloadcCsdjf|jS)NzTarget type: {target_type})format__dict__rQrrr extra_reprszCaltech101.extra_repr)rNNF)__name__ __module__ __qualname____doc__rr-rrrrboolr'intrrrMr.rSrr\ __classcell__rrrArr s$ )+rcszeZdZdZdeeeeeeddfdd Ze e e e fddd Z ed d d Z e d d dZdd ddZZS) Caltech256a`Caltech 256 `_ Dataset. Args: root (str or ``pathlib.Path``): Root directory of dataset where directory ``caltech256`` exists or will be saved to if download is set to True. 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 in root directory. If dataset is already downloaded, it is not downloaded again. NF)rrrrrc stjtj|d||dtj|jdd|r8||sHt dt t tj|jd|_ g|_ g|_t|j D]V\}}tddt tj|jd|D}|j td |d |j||gqzdS) NZ caltech256rTrr256_ObjectCategoriescSsg|]}|dr|qS)rE)endswith)ritemrrrrs z'Caltech256.__init__..r )r&r'r(r)r*r+rrr.r/r0r1r2r7r8r9r:r;r<)r=rrrrr>r?r@rArrr's"zCaltech256.__init__rCc Csttj|jd|j|j||j|ddd|j|dd}|j|}|j dk rd| |}|j dk rx| |}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is index of the target class. rer Z03d_rDrEN) r rFr(r)r*rr2r8r7rr)r=r7rJrKrrrrMs"     zCaltech256.__getitem__rNcCstjtj|jdS)NrerOrQrrrr.szCaltech256._check_integritycCs t|jSr!rRrQrrrrSszCaltech256.__len__cCs*|rtddStd|jddddS)NrTzAhttps://drive.google.com/file/d/1r6o0pSROcV1_VwT4oSjA2FBUSCWGuxLKz256_ObjectCategories.tarZ 67b4f42ca05d46448c6bb8ecd2220f6drUrXrQrrrrszCaltech256.download)NNF)r]r^r_r`r-rrrar'rbrrrMr.rSrrcrrrArrdsrd)r(os.pathpathlibrtypingrrrrrrZPILr utilsr r Zvisionr rrdrrrrs