kh@ r d dl mZ d dlmZmZmZmZmZ ddlm Z ddl
m Z m Z m
Z
mZ ddlmZ G d de Zy )
)Path)AnyCallableOptionalTupleUnion )default_loader)check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc e Zd ZdZdZddddZddd d
Zd d d d
efdee e
f de de e de e de
de ee e
f gef dd f fd
ZdefdZdedeeef fdZde fdZd Zd Zg d Z xZS )
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.tgz 52808999861908f626f3c1f4e79d11fa)zimagelabels.mat e0620be6f572b9609742df49c70aed4d)z setid.mat a5357ecc9cb78c4bef273ce3793fc85c)imagelabelsetidtrnidvalidtstidtrainvaltestr NFrootsplit transformtarget_transformdownloadloaderreturnc & t
| ||| t |dd | _ t | j
dz | _ | j dz | _ |r| j | j s t d ddl m } || j | j d d z d
}|| j | j j } || j | j d d z d
}
t! t# |
d
dz
j d } g | _ g | _ | D ]M } | j$ j) | | | j& j) | j d| ddz O || _ y )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_filesappendr$ )selfr r r! r" r# r$ r( set_ids image_idsr* image_id_to_labelimage_id __class__s [/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/datasets/flowers102.pyr. zFlowers102.__init__. sy EUV$UG5MN Om ; " / /% 7 MMO$$& ij j$$++doog.Fq.IIVZ[D , ,T[[ 9:AAC **T__W-Ea-HHUYZ F8,QU;V%V W X c , t | j S N)lenr; r= s rC __len__zFlowers102.__len__R s 4$$%%rD idxc | j | | j | }}| j | }| j r| j | }| j r| j | }||fS rF )r; r: r$ r! r" )r= rJ
image_filer r s rC __getitem__zFlowers102.__getitem__U se - -c 2DLL4EE
J' >>NN5)E ))%0Ee|rD c d| j S )Nzsplit=)r/ rH s rC
extra_reprzFlowers102.extra_repra s }%%rD c | j j r| j j sydD ]8 }| j | \ }}t t | j |z | r8 y y)NFr r T)r1 existsis_dirr5 r strr0 r= idfilenamemd5s rC r2 zFlowers102._check_integrityd sk ##**,1D1D1K1K1M$ B OOB /MHc"3t'8'88'C#DcJ rD c R | j ry t | j | j d d t | j
| j d d dD ]B }| j | \ }}t
| j |z t | j
| D y )Nr r r )rX rQ )r2 r _download_url_prefixr5 rT r0 r
rU s rC r# zFlowers102.downloadn s " $(()$//'*B1*E)F G !! "(+
% `B OOB /MHc 22X=s4CTCT?U[^ _ `rD )fz
pink primrosezhard-leaved pocket orchidzcanterbury bellsz sweet peazenglish marigoldz
tiger lilyz moon orchidzbird of paradise monkshoodz
globe thistle
snapdragonz colt's footz king proteaz
spear thistlez yellow irisz globe-flowerzpurple coneflowerz
peruvian lilyzballoon flowerzgiant white arum lilyz fire lilyzpincushion flower
fritillaryz
red gingerzgrape hyacinthz
corn poppyzprince of wales featherszstemless gentian artichokez
sweet william carnationz garden phloxzlove in the mistz
mexican asterzalpine sea hollyzruby-lipped cattleyaz cape flowerzgreat masterwortz
siam tulipz lenten rosezbarbeton daisydaffodilz
sword lily
poinsettiazbolero deep blue
wallflowermarigold buttercupz oxeye daisyzcommon dandelionpetuniaz
wild pansyprimula sunflower pelargoniumzbishop of llandaffgaurageraniumz
orange dahliazpink-yellow dahlia?zcautleya spicatazjapanese anemonezblack-eyed susan
silverbushzcalifornian poppy osteospermumz
spring crocusz bearded iris
windflowerz
tree poppygazaniaazaleaz
water lilyrosez thorn applez
morning gloryzpassion flowerlotusz toad lily anthurium
frangipaniclematishibiscus columbinez desert-rosez tree mallowmagnoliacyclamen
watercressz
canna lily hippeastrumzbee balmz ball mossfoxglove
bougainvilleacamelliamallowzmexican petuniabromeliazblanket flowerztrumpet creeperzblackberry lily)__name__
__module__ __qualname____doc__rZ r5 r6 r
r rT r r r boolr r. intrI r rM rO r2 r# classes
__classcell__)rB s @rC r r
s 6 PGHBJ
$GWEK
(,/3 4B" CI " " H% "
#8, "
" %T *+S01"
" H& &
s
uS#X
&C &
`gGrD r N)pathlibr typingr r r r r folderr
utilsr r r
r visionr r rD rC r s( 8 8 " ^ ^ !W WrD