a
h9 @ sh d dl mZ d dlmZmZmZmZ ddlmZ ddl m
Z
m Z m Z m
Z
ddlmZ G dd deZd S )
)Path)AnyCallableOptionalUnion )default_loader)check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc
s e Zd ZdZdZddddZddd d
Zd d d d
efee e
f e e e e e e
e ee e
f gef d d fdd
ZedddZeeeef dddZe dddZdd Zdd Zg d Z ZS )
Flowers102aC `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 or torch.Tensor, depends on the given loader,
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.
loader (callable, optional): A function to load an image given its path.
By default, it uses PIL as its image loader, but users could also pass in
``torchvision.io.decode_image`` for decoding image data into tensors directly.
z2https://www.robots.ox.ac.uk/~vgg/data/flowers/102/)z102flowers.tgzZ 52808999861908f626f3c1f4e79d11fa)zimagelabels.matZ e0620be6f572b9609742df49c70aed4d)z setid.matZ a5357ecc9cb78c4bef273ce3793fc85c)imagelabelsetidZtrnidZvalidZtstidtrainvaltestr NF)rootsplit transformtarget_transformdownloadloaderreturnc
s t j|||d t|dd| _t| jd | _| jd | _|rH| | sXt
dddl m } || j| j
d d d
d }|| j| j } || j| j
d d d
d }
tt|
d
d d} g | _g | _| D ]0} | j| | | j| jd| dd q|| _d S )N)r r r r z flowers-102ZjpgzHDataset not found or corrupted. You can use download=True to download itr )loadmatr T)Z
squeeze_mer labelsr Zimage_Z05dz.jpg)super__init__r _splitr r _base_folder_images_folderr _check_integrity RuntimeErrorZscipy.ior
_file_dict _splits_maptolistdict enumerate_labels _image_filesappendr )
selfr r r r r r r Zset_idsZ image_idsr Zimage_id_to_labelZimage_id __class__ M/var/www/auris/lib/python3.9/site-packages/torchvision/datasets/flowers102.pyr . s& zFlowers102.__init__)r c C s
t | jS N)lenr, r. r1 r1 r2 __len__R s zFlowers102.__len__)idxr c C sH | j | | j| }}| |}| jr0| |}| jr@| |}||fS r3 )r, r+ r r r )r. r7 Z
image_filer r r1 r1 r2 __getitem__U s
zFlowers102.__getitem__c C s d| j S )Nzsplit=)r! r5 r1 r1 r2
extra_repra s zFlowers102.extra_reprc C sN | j r| j sdS dD ],}| j| \}}tt| j| |s dS q dS )NFr r T)r# existsis_dirr& r strr" r. idfilenamemd5r1 r1 r2 r$ d s zFlowers102._check_integrityc C st | r d S t| j | jd d t| j| jd d d dD ],}| j| \}}t| j| t| j|d qBd S )Nr r r )rA r: )r$ r
_download_url_prefixr& r= r" r r> r1 r1 r2 r n s zFlowers102.download)fz
pink primrosezhard-leaved pocket orchidzcanterbury bellsz sweet peazenglish marigoldz
tiger lilyz moon orchidzbird of paradiseZ monkshoodz
globe thistleZ
snapdragonz colt's footz king proteaz
spear thistlez yellow irisz globe-flowerzpurple coneflowerz
peruvian lilyzballoon flowerzgiant white arum lilyz fire lilyzpincushion flowerZ
fritillaryz
red gingerzgrape hyacinthz
corn poppyzprince of wales featherszstemless gentianZ artichokez
sweet williamZ carnationz garden phloxzlove in the mistz
mexican asterzalpine sea hollyzruby-lipped cattleyaz cape flowerzgreat masterwortz
siam tulipz lenten rosezbarbeton daisyZdaffodilz
sword lilyZ
poinsettiazbolero deep blueZ
wallflowerZmarigoldZ buttercupz oxeye daisyzcommon dandelionZpetuniaz
wild pansyZprimulaZ sunflowerZ pelargoniumzbishop of llandaffZgauraZgeraniumz
orange dahliazpink-yellow dahlia?zcautleya spicatazjapanese anemonezblack-eyed susanZ
silverbushzcalifornian poppyZ osteospermumz
spring crocusz bearded irisZ
windflowerz
tree poppyZgazaniaZazaleaz
water lilyZrosez thorn applez
morning gloryzpassion flowerZlotusz toad lilyZ anthuriumZ
frangipaniZclematisZhibiscusZ columbinez desert-rosez tree mallowZmagnoliaZcyclamenZ
watercressz
canna lilyZ hippeastrumzbee balmz ball mossZfoxgloveZ
bougainvilleaZcamelliaZmallowzmexican petuniaZbromeliazblanket flowerztrumpet creeperzblackberry lily)__name__
__module__ __qualname____doc__rB r& r' r r r= r r r boolr r intr6 tupler8 r9 r$ r classes
__classcell__r1 r1 r/ r2 r
s6
$
r N)pathlibr typingr r r r folderr utilsr r
r r Zvisionr
r r1 r1 r1 r2 s