a
PSic @ sd d dl mZ d dlmZmZmZmZ d dlZddl m
Z
m Z m Z m
Z
ddlmZ G dd deZdS ) )Path)AnyTupleCallableOptionalN )check_integrity download_and_extract_archive download_urlverify_str_arg)
VisionDatasetc s e Zd ZdZdZddddZddd d
Zd eeee ee e
d d fdd
Z e dddZ
eeef dddZedddZdd Zdd Z ZS )
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 (string): 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 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.
z2https://www.robots.ox.ac.uk/~vgg/data/flowers/102/)z102flowers.tgzZ 52808999861908f626f3c1f4e79d11fa)zimagelabels.matZ e0620be6f572b9609742df49c70aed4d)z setid.matZ a5357ecc9cb78c4bef273ce3793fc85c)imagelabelsetidZtrnidvalidZtstidtrainvaltestr NF)rootsplit transformtarget_transformdownloadreturnc 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 qd S )N)r r r r z flowers-102jpgzHDataset not found or corrupted. You can use download=True to download itr )loadmatr T)Z
squeeze_mer labelsr image_Z05dz.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 Zset_idsZ image_idsr Zimage_id_to_labelZimage_id __class__ [/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/datasets/flowers102.pyr! + s$ zFlowers102.__init__)r c C s
t | jS )N)lenr. r0 r3 r3 r4 __len__L s zFlowers102.__len__c C sP | j | | j| }}tj|d}| jr8| |}| jrH| |}||fS )NRGB)r. r- PILImageopenconvertr r )r0 idx
image_filer r r3 r3 r4 __getitem__O s
zFlowers102.__getitem__c C s d| j S )Nzsplit=)r" r6 r3 r3 r4
extra_repr[ 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# r0 idfilenamemd5r3 r3 r4 r% ^ 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 )rH rA )r% r _download_url_prefixr( rD r# r
rE r3 r3 r4 r h s zFlowers102.download)r NNF)__name__
__module__ __qualname____doc__rI r( r) rD r r boolr! intr7 r r r? r@ r% r
__classcell__r3 r3 r1 r4 r
s0 !
r
)pathlibr typingr r r r PIL.Imager9 utilsr r r
r visionr r
r3 r3 r3 r4 s