%`i"ddlZddlZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZmZddlmZGdd eZGd d eZdS) N)Path)AnyCallableOptionalUnion)Image)download_and_extract_archiveextract_archiveverify_str_arg) VisionDatasetc eZdZdZ ddeeefdeeeefdee dee d e d df fd Z d e d e eeffd Zd e fdZd e fdZddZd efdZxZS) 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  ttj|d||tj|jdt|tr|g}d|D|_ |r| | stdttjtj|jd|_|jdd d d d d  t#t% fd|j|_g|_g|_t-|jD]\}}t/tjtj|jd|}|jt3d|dz|j||gzdS)N caltech101rrTexist_okc0g|]}t|ddS)r)r annotation)r ).0ts p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchvision/datasets/caltech.py z'Caltech101.__init__..5s&nnn]^N1m=WXXnnnHDataset not found or corrupted. You can use download=True to download it101_ObjectCategoriesBACKGROUND_GoogleFaces_2Faces_3 Motorbikes_16Airplanes_Side_2)Faces Faces_easy Motorbikes airplanesc |vr|n|SN)xname_maps r z%Caltech101.__init__..IsqH}} Z[r"r )super__init__ospathjoinmakedirsr isinstancestrrr_check_integrity RuntimeErrorsortedlistdir categoriesremovelistmapannotation_categoriesindexy enumeratelenextendrange) selfrrrrricnr2 __class__s @r r5zCaltech101.__init__)s dL99Yaqrrr DI---- k3 ' ' (&-Knnbmnnn   MMOOO$$&& kijj j BGLLDZ,[,[!\!\]] 2333 #)+    &*#.[.[.[.[]a]l*m*m%n%n" " do.. # #DAqBJrw||DI7MqQQRRSSA J  eAq1uoo . . . FMM!qc' " " " " # #r"rEc ddl}tjtj|jd|j|j|d|j |dd}g}|j D]}|dkr!| |j|)|dkr|j tj|jd |j|j|d |j |dd }| |d t|d krt!|n|d}|j||}|j||}||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.ioropenr6r7r8rr@rFrErappendioloadmatrDrHtuplerr)rKrEscipyimgtargetrdatas r __getitem__zCaltech101.__getitem__Rs~ j GLL &u .4E*4444     ! 3 3AJ dfUm,,,,l""x''GLL %246%=AAdj&7AAAA  d=1222"%f++//vvay > %..%%C  ,**622FF{r"ctjtj|jdS)Nr$r6r7existsr8rrKs r r<zCaltech101._check_integrity})w~~bgll496LMMNNNr"c*t|jSr/rHrEres r __len__zCaltech101.__len__4:r"c|rdStd|jddtj|jd}tj|D]J}|dr3ttj|||jKtj |tj tj|jddS)NzBhttps://data.caltech.edu/records/mzrjq-6wc02/files/caltech-101.zipzcaltech-101.zip 3138e1922a9193bfa496528edbbc45d0) download_rootfilenamemd5z caltech-101z.gz) r<r rr6r7r8r?endswithr shutilrmtreerA)rK gzip_folder gzip_files r rzCaltech101.downloads  " "  F$ P)&2     gll49m<< K00 Q QI!!%(( Q [) D DdiPPP k""" "',,ty*;<<=====r"c&djdi|jS)NzTarget type: {target_type}r0)format__dict__res r extra_reprzCaltech101.extra_reprs2+2CCT]CCCr")rNNFrN)__name__ __module__ __qualname____doc__rr;rrBrrboolr5intr\rrar<rirrx __classcell__rOs@r rr sS<.8(,/3 '#'#CI'#49c>*'#H% '# #8, '#  '# '#'#'#'#'#'#R))sCx))))VO$OOOO>>>>"DCDDDDDDDDr"rc eZdZdZ ddedeedeededdf fd Zd e de e e ffd Z defd Z de fd ZddZxZS) 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 Fttj|d||tj|jd|r||stdttj tj|jd|_ g|_ g|_t|j D]\}}t!dtj tj|jd|D}|j t%d|dz|j||gzdS) N caltech256rTrr#256_ObjectCategoriesc<g|]}|d|S)rS)rp)ritems r r!z'Caltech256.__init__..s:}}V,,r"r )r4r5r6r7r8r9rrr<r=r>r?r@rErFrGrHrIrJ) rKrrrrrLrMrNrOs r r5zCaltech256.__init__sr dL99Yaqrrr DI----   MMOOO$$&& kijj j BGLLDZ,[,[!\!\]] " do.. # #DAq " 27<< CY[\+]+] ^ ^A J  eAq1uoo . . . FMM!qc' " " " " # #r"rEc ttjtj|jd|j|j||j|dzdd|j|dd}|j|}|j | |}|j | |}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is index of the target class. rr 03d_rRrS) rrXr6r7r8rr@rFrErr)rKrEr^r_s r razCaltech256.__getitem__sj GLL &u .6%=1$FFF4:e+<FFFF      > %..%%C  ,**622FF{r"ctjtj|jdS)Nrrcres r r<zCaltech256._check_integrityrfr"c*t|jSr/rhres r rizCaltech256.__len__rjr"cb|rdStd|jdddS)NzKhttps://data.caltech.edu/records/nyy15-4j048/files/256_ObjectCategories.tarz256_ObjectCategories.tar 67b4f42ca05d46448c6bb8ecd2220f6d)rnro)r<r rres r rzCaltech256.downloadsJ  " "  F$ Y I/2       r")NNFry)rzr{r|r}r;rrr~r5rr\rrar<rirrrs@r rrs   ")-/3 ###H%##8, #  #  ######<sCx6O$OOOO         r"r)r6os.pathrqpathlibrtypingrrrrPILrutilsr r r visionr rrr0r"r rs  111111111111PPPPPPPPPP!!!!!!IDIDIDIDIDIDIDIDXX X X X X X X X X X r"