o ^i"@sddlZddlZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZmZddlmZGdd d eZGd d d eZdS) N)Path)AnyCallableOptionalUnion)Image)download_and_extract_archiveextract_archiveverify_str_arg) VisionDatasetc seZdZdZ    ddeeefdeeeefdee dee d e d df fd d Z d e d e eeffddZd e fddZd e fddZdddZd efddZZS) 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. 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)N caltech101rrTexist_okcSsg|]}t|ddqS)r)r annotation)r ).0trW/var/www/html/RAG/RAG_venv/lib/python3.10/site-packages/torchvision/datasets/caltech.py 5sz'Caltech101.__init__..HDataset not found or corrupted. You can use download=True to download it101_ObjectCategoriesBACKGROUND_GoogleFaces_2Faces_3 Motorbikes_16Airplanes_Side_2)Faces Faces_easy Motorbikes airplanescs|vr|S|SNr)xname_maprrIsz%Caltech101.__init__..r)super__init__ospathjoinmakedirsr isinstancestrrr_check_integrity RuntimeErrorsortedlistdir categoriesremovelistmapannotation_categoriesindexy enumeratelenextendrange) selfrrrrricn __class__r,rr0)s0  zCaltech101.__init__r@c 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. rNr image_04d.jpgrr Annotations annotation_z.mat obj_contourr)scipy.ioropenr1r2r3rr;rAr@rappendioloadmatr?rCtuplerr)rFr@scipyimgtargetrdatarrr __getitem__Rs<     zCaltech101.__getitem__cCtjtj|jdS)Nr r1r2existsr3rrFrrrr7}zCaltech101._check_integritycC t|jSr*rCr@r`rrr__len__ zCaltech101.__len__cCs|rdStd|jdddtj|jd}t|D]}|dr.ttj|||jqt |t tj|jddS)NzBhttps://data.caltech.edu/records/mzrjq-6wc02/files/caltech-101.zipzcaltech-101.zip 3138e1922a9193bfa496528edbbc45d0) download_rootfilenamemd5z caltech-101z.gz) r7r rr1r2r3r:endswithr shutilrmtreer<)rF gzip_folder gzip_filerrrrs  zCaltech101.downloadcCsdjdi|jS)NzTarget type: {target_type}r)format__dict__r`rrr extra_reprszCaltech101.extra_repr)rNNFrN)__name__ __module__ __qualname____doc__rr6rr=rrboolr0intrWrr\r7rdrrq __classcell__rrrJrr s0 )+ r c seZdZdZ   ddedeedeededdf fd d Zd e de e e ffd d Z defddZ de fddZdddZZS) 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. 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) N caltech256rTrr256_ObjectCategoriescSsg|] }|dr|qS)rN)rj)ritemrrrrs z'Caltech256.__init__..r)r/r0r1r2r3r4rrr7r8r9r:r;r@rArBrCrDrE)rFrrrrrGrHrIrJrrr0s$zCaltech256.__init__r@c 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. r|r03d_rMrNN) rrSr1r2r3rr;rAr@rr)rFr@rYrZrrrr\s"    zCaltech256.__getitem__cCr])Nr|r^r`rrrr7razCaltech256._check_integritycCrbr*rcr`rrrrdrezCaltech256.__len__cCs"|rdStd|jddddS)NzKhttps://data.caltech.edu/records/nyy15-4j048/files/256_ObjectCategories.tarz256_ObjectCategories.tar 67b4f42ca05d46448c6bb8ecd2220f6d)rhri)r7r rr`rrrrs zCaltech256.download)NNFrr)rsrtrurvr6rrrwr0rxrWrr\r7rdrryrrrJrrzs(rz)r1os.pathrkpathlibrtypingrrrrPILrutilsr r r visionr r rzrrrrs