U
    (d                     @   sd   d dl Zd dlmZmZmZmZ d dlZd dl	m
Z
 ddlmZmZ ddlmZ G dd deZdS )	    N)AnyCallableOptionalTuple)Image   )download_urlcheck_integrity)VisionDatasetc                       s   e Zd ZdZdZdZdZdeee	 ee	 e
dd fdd	Zeeeef d
ddZedddZe
dddZddddZ  ZS )SEMEIONa  `SEMEION <http://archive.ics.uci.edu/ml/datasets/semeion+handwritten+digit>`_ 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 cb545d371d2ce14ec121470795a77432NT)root	transformtarget_transformdownloadreturnc                    s   t  j|||d |r|   |  s.tdtj| j| j	}t
|}|d d d df d d| _t
| jd| _t
|d d dd f d | _d S )N)r   r   zHDataset not found or corrupted. You can use download=True to download it      Zuint8)   r   r   )super__init__r   _check_integrityRuntimeErrorospathjoinr   filenamenpZloadtxtZastypedataZreshapeZnonzerolabels)selfr   r   r   r   fpr   	__class__ @/tmp/pip-unpacked-wheel-vx7f76es/torchvision/datasets/semeion.pyr      s    
 zSEMEION.__init__)indexr   c                 C   sX   | j | t| j|  }}tj|dd}| jdk	r<| |}| jdk	rP| |}||fS )z
        Args:
            index (int): Index

        Returns:
            tuple: (image, target) where target is index of the target class.
        L)modeN)r   intr   r   Z	fromarrayr   r   )r    r&   imgtargetr$   r$   r%   __getitem__5   s    



zSEMEION.__getitem__)r   c                 C   s
   t | jS )N)lenr   )r    r$   r$   r%   __len__K   s    zSEMEION.__len__c                 C   s*   | j }tj|| j}t|| js&dS dS )NFT)r   r   r   r   r   r	   md5_checksum)r    r   Zfpathr$   r$   r%   r   N   s
    zSEMEION._check_integrityc                 C   s2   |   rtd d S | j}t| j|| j| j d S )Nz%Files already downloaded and verified)r   printr   r   urlr   r/   )r    r   r$   r$   r%   r   U   s
    zSEMEION.download)NNT)__name__
__module____qualname____doc__r1   r   r/   strr   r   boolr   r)   r   r   r,   r.   r   r   __classcell__r$   r$   r"   r%   r      s$      r   )os.pathr   typingr   r   r   r   Znumpyr   ZPILr   utilsr   r	   Zvisionr
   r   r$   r$   r$   r%   <module>   s   