B \,@sdZddlmZmZmZddlmZmZddlZ ddl m Z ddl m Z ddl mZdd l mZdd l mZd d lmZd d lmZmZeddddZdddZdS)aModified Olivetti faces dataset. The original database was available from (now defunct) https://www.cl.cam.ac.uk/research/dtg/attarchive/facedatabase.html The version retrieved here comes in MATLAB format from the personal web page of Sam Roweis: https://cs.nyu.edu/~roweis/ )dirnameexistsjoin)makedirsremoveN)loadmat) get_data_home) _fetch_remote)RemoteFileMetadata) _pkl_filepath)_joblib)check_random_stateBunchzolivettifaces.matz.https://ndownloader.figshare.com/files/5976027Z@b612fb967f2dc77c9c62d3e1266e0c73d5fca46a4b8906c18e454d41af987794)filenameurlZchecksumFTc CsVt|d}t|st|t|d}t|s|s8tdtdtj|ftt|d}t |d}t ||dj }t j||dd ~n t |}t|}||}||}|d d d d }tddtdD}|rt|}|t|} || }|| }tt} tt| dd} | } WdQRXt |t|d||| dS)aELoad the Olivetti faces data-set from AT&T (classification). Download it if necessary. ================= ===================== Classes 40 Samples total 400 Dimensionality 4096 Features real, between 0 and 1 ================= ===================== Read more in the :ref:`User Guide `. Parameters ---------- data_home : optional, default: None Specify another download and cache folder for the datasets. By default all scikit-learn data is stored in '~/scikit_learn_data' subfolders. shuffle : boolean, optional If True the order of the dataset is shuffled to avoid having images of the same person grouped. random_state : int, RandomState instance or None (default=0) Determines random number generation for dataset shuffling. Pass an int for reproducible output across multiple function calls. See :term:`Glossary `. download_if_missing : optional, True by default If False, raise a IOError if the data is not locally available instead of trying to download the data from the source site. Returns ------- An object with the following attributes: data : numpy array of shape (400, 4096) Each row corresponds to a ravelled face image of original size 64 x 64 pixels. images : numpy array of shape (400, 64, 64) Each row is a face image corresponding to one of the 40 subjects of the dataset. target : numpy array of shape (400, ) Labels associated to each face image. Those labels are ranging from 0-39 and correspond to the Subject IDs. DESCR : string Description of the modified Olivetti Faces Dataset. ) data_homez olivetti.pkzz1Data not found and `download_if_missing` is Falsez(downloading Olivetti faces from %s to %s)r) file_namefaces)compress)i@rrr rcSsg|] }|dqS) ).0irr>lib/python3.7/site-packages/sklearn/datasets/olivetti_faces.py wsz(fetch_olivetti_faces..iZdescrzolivetti_faces.rstN)dataZimagestargetZDESCR)!r rrr IOErrorprintFACESrr rrTcopyrdumploadnpZfloat32minmaxZreshapeZ transposeZarrayrangerZ permutationlenr__file__openrreadr) rZshuffleZ random_stateZdownload_if_missingfilepathZmat_pathZmfilerr!orderZ module_pathZrst_fileZfdescrrrrfetch_olivetti_faces&sB5        r3)NFrT)__doc__os.pathrrrosrrZnumpyr)Zscipy.io.matlabrbaser r r r Zutilsrrrr$r3rrrr s