Ph/ddlZddlZddlmZmZmZmZmZmZm Z m Z ddl m Z ddl mZdede ee edffd efd Zded efd Zd ed e eeeeefffd Z dd edeeeefdee ee edffdeeegefd ee eeff dZGddeZdZded e jfdZded efdZded efdZGddeZy)N)AnyCallablecastDictListOptionalTupleUnion)Image) VisionDatasetfilename extensions.returnc~|jjt|tr|St |S)aChecks if a file is an allowed extension. Args: filename (string): path to a file extensions (tuple of strings): extensions to consider (lowercase) Returns: bool: True if the filename ends with one of given extensions )lowerendswith isinstancestrtuple)rrs fC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/folder.pyhas_file_allowed_extensionr s4 >>  $ $:j#3NZ ffTYZdTe ffc"t|tS)zChecks if a file is an allowed image extension. Args: filename (string): path to a file Returns: bool: True if the filename ends with a known image extension )rIMG_EXTENSIONS)rs r is_image_filers &h ??r directoryctdtj|D}|std|dt |Dcic]\}}|| }}}||fScc}}w)zWFinds the class folders in a dataset. See :class:`DatasetFolder` for details. c3VK|]!}|js|j#ywN)is_dirname).0entrys r zfind_classes..(sU-BEellnUZZ-Bs))z"Couldn't find any class folder in .)sortedosscandirFileNotFoundError enumerate)rclassesicls_name class_to_idxs r find_classesr0#si URZZ -BUUG "DYKq QRR3.is_valid_fileKs-a< MI|0023 "<0 WW\\)\: ww}}Z( %bggjd&KLOD!Vww||D%0 &,D$$T*#+<<)--l;( M 4 ))+,/@@M4TYYvm?T5U4VVXY  ! /jUX>Y _c_h_his_t/uv vC$$ rc<eZdZdZ ddedeegefdeeedfdeedeed eeege fd dffd Z e dd ed e ee fdeeedfd eeege fd eeee ff dZd ed eeee ee fffdZde d eeeffdZd e fdZxZS) DatasetFolderaA generic data loader. This default directory structure can be customized by overriding the :meth:`find_classes` method. Args: root (string): Root directory path. loader (callable): A function to load a sample given its path. extensions (tuple[string]): A list of allowed extensions. both extensions and is_valid_file should not be passed. transform (callable, optional): A function/transform that takes in a sample and returns a transformed version. E.g, ``transforms.RandomCrop`` for images. target_transform (callable, optional): A function/transform that takes in the target and transforms it. is_valid_file (callable, optional): A function that takes path of a file and check if the file is a valid file (used to check of corrupt files) both extensions and is_valid_file should not be passed. Attributes: classes (list): List of the class names sorted alphabetically. class_to_idx (dict): Dict with items (class_name, class_index). samples (list): List of (sample path, class_index) tuples targets (list): The class_index value for each image in the dataset NrIloaderr. transformtarget_transformr1rc"t |||||j|j\}}|j |j|||} ||_||_||_||_| |_ | D cgc]} | d c} |_ ycc} w)N)rSrTr ) super__init__r0rIrOrRrr,r/samplestargets) selfrIrRrrSrTr1r,r/rXs __class__s rrWzDatasetFolder.__init__s EUV $ 1 1$)) <##DII|ZW $ ( &-.g!g. .s7 B rr/c:| tdt||||S)anGenerates a list of samples of a form (path_to_sample, class). This can be overridden to e.g. read files from a compressed zip file instead of from the disk. Args: directory (str): root dataset directory, corresponding to ``self.root``. class_to_idx (Dict[str, int]): Dictionary mapping class name to class index. extensions (optional): A list of allowed extensions. Either extensions or is_valid_file should be passed. Defaults to None. is_valid_file (optional): A function that takes path of a file and checks if the file is a valid file (used to check of corrupt files) both extensions and is_valid_file should not be passed. Defaults to None. Raises: ValueError: In case ``class_to_idx`` is empty. ValueError: In case ``extensions`` and ``is_valid_file`` are None or both are not None. FileNotFoundError: In case no valid file was found for any class. Returns: List[Tuple[str, int]]: samples of a form (path_to_sample, class) z*The class_to_idx parameter cannot be None.)rr1)r8rO)rr/rr1s rrOzDatasetFolder.make_datasets*:  IJ JI| Zghhrct|S)utFind the class folders in a dataset structured as follows:: directory/ ├── class_x │ ├── xxx.ext │ ├── xxy.ext │ └── ... │ └── xxz.ext └── class_y ├── 123.ext ├── nsdf3.ext └── ... └── asd932_.ext This method can be overridden to only consider a subset of classes, or to adapt to a different dataset directory structure. Args: directory(str): Root directory path, corresponding to ``self.root`` Raises: FileNotFoundError: If ``dir`` has no class folders. Returns: (Tuple[List[str], Dict[str, int]]): List of all classes and dictionary mapping each class to an index. )r0)rZrs rr0zDatasetFolder.find_classess6I&&rindexc|j|\}}|j|}|j|j|}|j|j|}||fS)z Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class. )rXrRrSrT)rZr_r6targetsamples r __getitem__zDatasetFolder.__getitem__sb||E* fT" >> %^^F+F  ,**62Fv~rc,t|jSr )lenrX)rZs r__len__zDatasetFolder.__len__s4<<  r)NNNN)NN)__name__ __module__ __qualname____doc__rrrrr r9rW staticmethodrintrrOr0rcrf __classcell__r\s@rrQrQkse<15(,/39=//#$/U38_- / H% / #8, / # 56/ /*159= !i!i38n!iU38_-!i # 56 !i eCHo  !i!iF'c'eDItCH~4M.N':sCx"!!rrQ) z.jpgz.jpegz.pngz.ppmz.bmpz.pgmz.tifz.tiffz.webpr6ct|d5}tj|}|jdcdddS#1swYyxYw)NrbRGB)openr convert)r6fimgs r pil_loaderrvs3 dD Qjjm{{5!   s &=Acbddl} |j|S#t$rt|cYSwxYw)Nr)accimager OSErrorrv)r6rxs raccimage_loaderrzs3 ~~d##  $ s ..cNddlm}|dk(r t|St|S)Nr)get_image_backendrx) torchvisionr|rzrv)r6r|s rdefault_loaderr~s&-j(t$$$rcleZdZdZddedfdedeedeedeegefdeeege ff fd Z xZ S) ImageFolderaA generic data loader where the images are arranged in this way by default: :: root/dog/xxx.png root/dog/xxy.png root/dog/[...]/xxz.png root/cat/123.png root/cat/nsdf3.png root/cat/[...]/asd932_.png This class inherits from :class:`~torchvision.datasets.DatasetFolder` so the same methods can be overridden to customize the dataset. Args: root (string): Root directory path. 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. loader (callable, optional): A function to load an image given its path. is_valid_file (callable, optional): A function that takes path of an Image file and check if the file is a valid file (used to check of corrupt files) Attributes: classes (list): List of the class names sorted alphabetically. class_to_idx (dict): Dict with items (class_name, class_index). imgs (list): List of (image path, class_index) tuples NrIrSrTrRr1cbt||||tnd||||j|_y)N)rSrTr1)rVrWrrXimgs)rZrIrSrTrRr1r\s rrWzImageFolder.__init__-s=   +3N-'  LL r) rgrhrirjr~rrrrr9rWrmrns@rrrsv@)-/3'59= !!H%!#8, ! #$ !  # 56 !!rr)NNN)r(os.pathtypingrrrrrrr r PILr visionr rr9rrrlr0rOrQrrvrzr~rrrrsk JJJ! g g%U3PS8_@T:U gZ^ g @C @D @ !C !E$s)T#s(^*C$D !.28<59 884S>*8sE#s(O3458HcUD[12 8  %S/ 8vC!MC!L]"S"U[[" # #    .!-.!r