U
h @ st d dl Z d dlZ d dlZd dlmZmZmZmZmZm Z d dl
m Z ddl m
Z
mZ ddlmZ G dd deZdS ) N)AnyCallableOptionalSequenceTupleUnion)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc s e Zd ZdZdZdZdeeej f eee
e ef e e e e e e e
d fd d
Zed d d
Zeeeef dddZe
d ddZdd ddZ ZS )
OxfordIIITPeta `Oxford-IIIT Pet Dataset `_.
Args:
root (str or ``pathlib.Path``): Root directory of the dataset.
split (string, optional): The dataset split, supports ``"trainval"`` (default) or ``"test"``.
target_types (string, sequence of strings, optional): Types of target to use. Can be ``category`` (default) or
``segmentation``. Can also be a list to output a tuple with all specified target types. The types represent:
- ``category`` (int): Label for one of the 37 pet categories.
- ``binary-category`` (int): Binary label for cat or dog.
- ``segmentation`` (PIL image): Segmentation trimap of the image.
If empty, ``None`` will be returned as target.
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 into
``root/oxford-iiit-pet``. If dataset is already downloaded, it is not downloaded again.
))z=https://www.robots.ox.ac.uk/~vgg/data/pets/data/images.tar.gzZ 5c4f3ee8e5d25df40f4fd59a7f44e54c)zBhttps://www.robots.ox.ac.uk/~vgg/data/pets/data/annotations.tar.gzZ 95a8c909bbe2e81eed6a22bccdf3f68f)categorybinary-categoryZ segmentationtrainvalr NF)rootsplit target_types
transforms transformtarget_transformdownloadc s t |dd _t|tr |g} fdd|D _t j||||d t j d _
j
d _ j
d _ j d _
|r std
g }g _g _t j j d V} | D ]J}
|
\} } }
}|| jt| d jt|
d qW 5 Q R X d
dg _dd tdd t| jD dd dD _t t jt t j _ t t jt t j _ fdd|D _! fdd|D _"d S )Nr )r testc s g | ]}t |d jqS )r )r _VALID_TARGET_TYPES).0 target_typeself V/var/www/html/venv/lib/python3.8/site-packages/torchvision/datasets/oxford_iiit_pet.py
5 s z*OxfordIIITPet.__init__..)r r r zoxford-iiit-petZimages annotationsZtrimapsz;Dataset not found. You can use download=True to download itz.txtr ZCatZDogc S s* g | ]"\}}d dd |dD qS ) c s s | ]}| V qd S N)title)r partr r r Q s z4OxfordIIITPet.__init__..._)joinr )r Zraw_clsr' r r r r P s c S s$ h | ] \}}| d dd |fqS )r' r r )rsplit)r image_idlabelr r r S s z)OxfordIIITPet.__init__..c S s | d S )Nr r )Zimage_id_and_labelr r r T z(OxfordIIITPet.__init__..)keyc s g | ]} j | d qS )z.jpg)_images_folderr r* r r r r Z s c s g | ]} j | d qS )z.png) _segs_folderr1 r r r r [ s )#r _split
isinstancestr
_target_typessuper__init__pathlibPathr _base_folderr0 _anns_folderr2 _download
_check_exists RuntimeError_labels _bin_labelsopenstripr appendintZ bin_classessortedzipclassesdictrangelenZbin_class_to_idxZ class_to_idx_images_segs)r r r r r r r r Z image_idsfileliner* r+ Z bin_labelr' __class__r r r8 ( sF
zOxfordIIITPet.__init__)returnc C s
t | jS r# )rK rL r r r r __len__] s zOxfordIIITPet.__len__)idxrR c C s t | j| d}g }| jD ]N}|dkr>|| j| q |dkrX|| j| q |t | j| q |szd }n t |dkr|d }nt
|}| j r| ||\}}||fS )NRGBr r r r ) r rB rL convertr6 rD r@ rA rM rK tupler )r rT imagetargetr r r r __getitem__` s
zOxfordIIITPet.__getitem__c C s4 | j | jfD ]"}tj|r(tj|s dS q dS )NFT)r0 r< ospathexistsisdir)r folderr r r r> x s zOxfordIIITPet._check_existsc C s4 | r d S | jD ] \}}t|t| j|d qd S )N)Z
download_rootmd5)r>
_RESOURCESr
r5 r; )r urlr` r r r r= s zOxfordIIITPet._download)r r NNNF)__name__
__module__ __qualname____doc__ra r r r5 r9 r: r r r boolr8 rE rS r r rZ r> r=
__classcell__r r rP r r
s, 5r
)r[ os.pathr9 typingr r r r r r ZPILr utilsr
r Zvisionr r
r r r r s