Di"ddlZddlZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZmZddlmZGdd eZGd d eZy) N)Path)AnyCallableOptionalUnion)Image)download_and_extract_archiveextract_archiveverify_str_arg) VisionDatasetc eZdZdZ ddeeefdeeeefdee dee de ddf fd Z d e de eeffd Zde fd Zde fd ZddZdefdZxZS) 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. Nroot target_type transformtarget_transformdownloadreturnc  t |tjj |d||tj |j dt|tr|g}|Dcgc]}t|ddc}|_ |r|j|js tdttjtjj |j d|_|j j#d d d d d d t%t' fd|j |_g|_g|_t/|j D]\}}t1tjtjj |j d|} |j*j3t5d| dz|j,j3| |gzycc}w)N caltech101rrTexist_okr)category annotationHDataset 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|S|SN)xname_maps ^/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/caltech.pyz%Caltech101.__init__..IsqH} Z[r )super__init__ospathjoinmakedirsr isinstancestrr rr_check_integrity RuntimeErrorsortedlistdir categoriesremovelistmapannotation_categoriesindexy enumeratelenextendrange) selfrrrrrticnr, __class__s @r-r1zCaltech101.__init__)s dL9Yaqr DII- k3 '&-Kbmn]^N1m=WXn  MMO$$&ij j BGGLLDZ,[!\] 23 #)+   &*#.[]a]l]l*m%n" " doo. #DAqBJJrww||DII7MqQRSA JJ  eAq1uo . FFMM!qc' " #1os+HrAc ddl}tjtjj |j d|j|j|d|j|dd}g}|jD]}|dk(r|j|j|'|dk(s-|jjtjj |j d |j|j|d |j|dd }|j|d t|d kDr t!|n|d}|j"|j#|}|j$|j%|}||fS)z Args: index (int): Index Returns: tuple: (image, target) where the type of target specified by target_type. rNrimage_04d.jpgrr Annotations annotation_z.mat obj_contourr )scipy.ioropenr2r3r4rr<rBrArappendioloadmatr@rDtuplerr)rGrAscipyimgtargetrHdatas r- __getitem__zCaltech101.__getitem__Rsb jj GGLL &u .E*3/t4   !! 3AJ dffUm,l"xx''GGLL %22466%=A%djj&7%> %..%C  ,**62FF{r/ctjjtjj|jdS)Nrr2r3existsr4rrGs r-r8zCaltech101._check_integrity})ww~~bggll4996LMNNr/c,t|jSr)rDrArbs r-__len__zCaltech101.__len__4::r/c|jrytd|jddtjj |jd}tj |D]H}|jdsttjj |||jJtj|tjtjj |jdy)NzBhttps://data.caltech.edu/records/mzrjq-6wc02/files/caltech-101.zipzcaltech-101.zip 3138e1922a9193bfa496528edbbc45d0) download_rootfilenamemd5z caltech-101z.gz) r8r rr2r3r4r;endswithr shutilrmtreer=)rG gzip_folder gzip_files r-rzCaltech101.downloads  " $ P))&2  ggll499m< K0 QI!!%( [) DdiiP Q  k" "'',,tyy*;<=r/c:djdi|jS)NzTarget type: {target_type}r*)format__dict__rbs r- extra_reprzCaltech101.extra_reprs2+22CT]]CCr/)rNNFrN)__name__ __module__ __qualname____doc__rr7rr>rrboolr1intrYrr^r8rfrru __classcell__rLs@r-rr s<.8(,/3 '#CI'#49c>*'#H% '# #8, '#  '# '#R))sCx)VO$O>"DCDr/rc eZdZdZ ddedeedeededdf fd Zd e de e e ffd Z defd Z de fd Zdd ZxZS) 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. Nrrrrrc Rt |tjj |d||tj |j d|r|j|js tdttjtjj |j d|_ g|_ g|_t|jD]\}}t!tjtjj |j d|Dcgc]}|j#dr|c}}|jj%t'd|dz|jj%||gzycc}w) N caltech256rTrr256_ObjectCategoriesrPr )r0r1r2r3r4r5rrr8r9r:r;r<rArBrCrDrmrErF) rGrrrrrIrJitemrKrLs r-r1zCaltech256.__init__s9 dL9Yaqr DII-  MMO$$&ij j BGGLLDZ,[!\] " doo. #DAq!# 277<< CY[\+] ^}}V,A JJ  eAq1uo . FFMM!qc' " #s9F$ rAc tjtjj |j d|j |j||j|dzdd|j|dd}|j|}|j|j|}|j|j|}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is index of the target class. rr 03d_rOrP) rrUr2r3r4rr<rBrArr)rGrAr[r\s r-r^zCaltech256.__getitem__sjj GGLL &u .66%=1$S)4::e+> %..%C  ,**62FF{r/ctjjtjj|jdS)Nrr`rbs r-r8zCaltech256._check_integrityrcr/c,t|jSr)rerbs r-rfzCaltech256.__len__rgr/cX|jrytd|jddy)NzKhttps://data.caltech.edu/records/nyy15-4j048/files/256_ObjectCategories.tarz256_ObjectCategories.tar 67b4f42ca05d46448c6bb8ecd2220f6d)rkrl)r8r rrbs r-rzCaltech256.downloads)  " $ Y II/2  r/)NNFrv)rwrxryrzr7rrr{r1r|rYrr^r8rfrr}r~s@r-rrs ")-/3 ##H%##8, #  #  #<sCx6O$O  r/r)r2os.pathrnpathlibrtypingrrrrPILrutilsr r r visionr rrr*r/r-rsB  11PP!IDIDXX X r/