j v d dl mZ d dlmZmZmZmZmZ d dlZ ddl
m Z m Z m
Z
mZ ddlmZ G d de ZdS ) )Path)AnyCallableOptionalTupleUnionN )check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc
e Zd ZdZdZddddZddd d
Z ddeee f dede
e de
e de dd f fd
Z
defdZdedeeef fdZdefdZd Zd Z xZS )
Flowers102a `Oxford 102 Flower `_ Dataset.
.. warning::
This class needs `scipy `_ to load target files from `.mat` format.
Oxford 102 Flower is an image classification dataset consisting of 102 flower categories. The
flowers were chosen to be flowers commonly occurring in the United Kingdom. Each class consists of
between 40 and 258 images.
The images have large scale, pose and light variations. In addition, there are categories that
have large variations within the category, and several very similar categories.
Args:
root (str or ``pathlib.Path``): Root directory of the dataset.
split (string, optional): The dataset split, supports ``"train"`` (default), ``"val"``, or ``"test"``.
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.
z2https://www.robots.ox.ac.uk/~vgg/data/flowers/102/)z102flowers.tgz 52808999861908f626f3c1f4e79d11fa)zimagelabels.mat e0620be6f572b9609742df49c70aed4d)z setid.mat a5357ecc9cb78c4bef273ce3793fc85c)imagelabelsetidtrnidvalidtstidtrainvaltestr NFrootsplit transformtarget_transformdownloadreturnc ` t ||| t |dd | _ t | j dz | _ | j dz | _ |r| | st d ddl m } || j | j
d d z d
}|| j | j } || j | j
d d z d
} t! t# | d
dz
d }
g | _ g | _ |D ]I} | j |
| | j | j d| ddz Jd S )N)r r! r r z flowers-102jpgzHDataset not found or corrupted. You can use download=True to download itr )loadmatr T)
squeeze_mer labelsr image_05dz.jpg)super__init__r
_splitr r _base_folder_images_folderr" _check_integrity RuntimeErrorscipy.ior&
_file_dict _splits_maptolistdict enumerate_labels _image_filesappend)
selfr r r r! r" r& set_ids image_idsr( image_id_to_labelimage_id __class__s
PC:\Projects\Historical\venv\Lib\site-packages\torchvision/datasets/flowers102.pyr, zFlowers102.__init__+ s EUVVV$UG5MNN OOm ; " /% 7 MMOOO$$&& k ijj j$$$$$$'$+dog.Fq.IIVZ[[[D ,T[ 9:AACC *T_W-Ea-HHUYZZZ F8, *NN5))E 1))%00Ee|rB c d| j S )Nzsplit=)r- rE s rA
extra_reprzFlowers102.extra_repr[ s % %%%rB c | j r| j sdS dD ]:}| j | \ }}t t | j |z | s dS ;dS )NFr r T)r/ existsis_dirr3 r
strr. r; idfilenamemd5s rA r0 zFlowers102._check_integrity^ s #**,, 1D1K1K1M1M 5$ B OB /MHc"3t'88'C#D#DcJJ
uu
trB c R | rd S t | j | j d d t | j | j d d dD ]>}| j | \ }}t
| j |z t | j | ?d S )Nr r r )rZ rS )r0 r _download_url_prefixr3 rV r. r rW s rA r" zFlowers102.downloadh s " " F$( G$/'*B1*E G G ! " "(+
% ` `B OB /MHc 2X=s4CT?U?U[^ _ _ _ _ _ ` `rB )r NNF)__name__
__module__ __qualname____doc__r\ r3 r4 r rV r r r boolr, intrF r r rO rQ r0 r"
__classcell__)r@ s @rA r r
si 0 PGHB J
$GWEEK
(,/3
X XCI X X H% X
#8, X
X
X X X X X XB& & & & &
s
uS#X
&C & & & &
`
`
`
`
`
`
`rB r )pathlibr typingr r r r r PIL.ImagerJ utilsr
r r r
visionr r rB rA rj s 8 8 8 8 8 8 8 8 8 8 8 8 8 8 ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ! ! ! ! ! !h` h` h` h` h` h` h` h` h` h`rB