o )i""@s|ddlZddlZddlmZmZmZmZmZmZddl m Z ddl m Z m Z ddlmZGdddeZGd d d eZdS) N)AnyCallableListOptionalTupleUnion)Image)download_and_extract_archiveverify_str_arg) VisionDatasetc seZdZdZ    ddedeeeefdeedeed e d df fd d Z d e d e e e ffddZd e fddZd e fddZdddZd efddZZS) Caltech101a`Caltech 101 `_ Dataset. .. warning:: This class needs `scipy `_ to load target files from `.mat` format. Args: root (string): 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 an 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. categoryNFroot target_type transformtarget_transformdownloadreturnc s tjtj|d||dtj|jddt|tr|g}dd|D|_ |r,| | s4t dt ttj|jd|_|jd d d d d dttfdd|j|_g|_g|_t|jD]&\}}tttj|jd|}|jtd|d|j||gqgdS)NZ caltech101rrTexist_okcSsg|]}t|ddqS)r)r annotation)r ).0trk/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/datasets/caltech.py /sz'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|vr|S|SNr)xZname_maprrCsz%Caltech101.__init__..r )super__init__ospathjoinmakedirsr isinstancestrrr_check_integrity RuntimeErrorsortedlistdir categoriesremovelistmapannotation_categoriesindexy enumeratelenextendrange) selfrrrrricn __class__r"rr%#s0  zCaltech101.__init__r5c Csddl}ttj|jd|j|j|d|j |dd}g}|j D]6}|dkr3| |j|q$|dkrZ|j tj|jd |j|j|d |j |dd }| |d q$t|d kret|n|d}|jdurs||}|jdur}||}||fS)z Args: index (int): Index Returns: tuple: (image, target) where the type of target specified by target_type. rNrZimage_04d.jpgrrZ AnnotationsZ annotation_z.matZ obj_contourr )Zscipy.ioropenr&r'r(rr0r6r5rappendioZloadmatr4r8tuplerr)r;r5Zscipyimgtargetrdatarrr __getitem__Ls<     zCaltech101.__getitem__cCtjtj|jdS)Nrr&r'existsr(rr;rrrr,wzCaltech101._check_integritycC t|jSr r8r5rNrrr__len__{ zCaltech101.__len__cCs<|r tddStd|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 rrNrrrr~s zCaltech101.downloadcCsdjdi|jS)NzTarget type: {target_type}r)format__dict__rNrrr extra_reprszCaltech101.extra_repr)rNNFrN)__name__ __module__ __qualname____doc__r+rrrrboolr%intrrrJr,rRrr\ __classcell__rrr?rr s0)+ r c seZdZdZ   ddedeedeededdf fd d Zd e de e e ffd d Z defddZ de fddZdddZZS) Caltech256a`Caltech 256 `_ Dataset. Args: root (string): 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 an 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. NFrrrrrc stjtj|d||dtj|jdd|r||s$t dt t tj|jd|_ g|_ g|_t|j D]+\}}tddt tj|jd|D}|j td |d |j||gq=dS) NZ caltech256rTrr256_ObjectCategoriescSsg|] }|dr|qS)rB)endswith)ritemrrrrs z'Caltech256.__init__..r )r$r%r&r'r(r)rrr,r-r.r/r0r5r6r7r8r9r:)r;rrrrr<r=r>r?rrr%s$zCaltech256.__init__r5c Csttj|jd|j|j||j|ddd|j|dd}|j|}|j dur2| |}|j dur<| |}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is index of the target class. rfr Z03d_rArBN) rrCr&r'r(rr0r6r5rr)r;r5rGrHrrrrJs"    zCaltech256.__getitem__cCrK)NrfrLrNrrrr,rOzCaltech256._check_integritycCrPr rQrNrrrrRrSzCaltech256.__len__cCs*|r tddStd|jddddS)NrTzAhttps://drive.google.com/file/d/1r6o0pSROcV1_VwT4oSjA2FBUSCWGuxLKz256_ObjectCategories.tarZ 67b4f42ca05d46448c6bb8ecd2220f6drUrXrNrrrrs zCaltech256.download)NNFr])r^r_r`rar+rrrbr%rcrrrJr,rRrrdrrr?rres(re)r&Zos.pathtypingrrrrrrZPILrutilsr r Zvisionr r rerrrrs