o )i @sdddlZddlmZmZmZmZddlZddl m Z ddl m Z m Z ddlmZGdddeZdS) N)AnyCallableOptionalTuple)Image)check_integrity download_url) VisionDatasetc seZdZdZdZdZdZ   ddedee d ee d e d df fd d Z de d e eeffddZd e fddZd e fddZdddZZS)SEMEIONa`SEMEION `_ Dataset. Args: root (string): Root directory of dataset where directory ``semeion.py`` exists. 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. zMhttp://archive.ics.uci.edu/ml/machine-learning-databases/semeion/semeion.dataz semeion.dataZ cb545d371d2ce14ec121470795a77432NTroot transformtarget_transformdownloadreturncstj|||d|r||stdtj|j|j }t |}|ddddfd d|_ t |j d|_ t |ddddfd|_dS)N)r rzHDataset not found or corrupted. You can use download=True to download itZuint8)rr)super__init__r_check_integrity RuntimeErrorospathjoinr filenamenpZloadtxtZastypedataZreshapeZnonzerolabels)selfr r rrfpr __class__k/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/datasets/semeion.pyrs  $zSEMEION.__init__indexcCsX|j|t|j|}}tj|dd}|jdur||}|jdur(||}||fS)z Args: index (int): Index Returns: tuple: (image, target) where target is index of the target class. L)modeN)rintrrZ fromarrayr r)r r&Zimgtargetr$r$r% __getitem__5s    zSEMEION.__getitem__cCs t|jS)N)lenr)r r$r$r%__len__Ks zSEMEION.__len__cCs*|j}tj||j}t||jsdSdS)NFT)r rrrrr md5_checksum)r r Zfpathr$r$r%rNs  zSEMEION._check_integritycCs2|r tddS|j}t|j||j|jdS)Nz%Files already downloaded and verified)rprintr r urlrr.)r r r$r$r%rUs zSEMEION.download)NNT)rN)__name__ __module__ __qualname____doc__r0rr.strrrboolrr)rrr+r-rr __classcell__r$r$r"r%r s.r )Zos.pathrtypingrrrrnumpyrZPILrutilsrr Zvisionr r r$r$r$r%s