o
V 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 ZdZdZdZ ddeeej f d ed
ee
e ef d e e d e e d
e e de
f fdd
ZdefddZdedeeef fddZde
fddZd 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.
transforms (callable, optional): A function/transform that takes input sample
and its target as entry and returns a transformed version.
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.gz 5c4f3ee8e5d25df40f4fd59a7f44e54c)zBhttps://www.robots.ox.ac.uk/~vgg/data/pets/data/annotations.tar.gz 95a8c909bbe2e81eed6a22bccdf3f68f)categorybinary-category segmentationtrainvalr NFrootsplit 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 _
|rD sLtd
g }g _g _t j j d 0} | D ]%}
|
\} } }
}|| jt| d jt|
d qbW d n1 sw Y 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 X/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/datasets/oxford_iiit_pet.py
7 s z*OxfordIIITPet.__init__..)r r r zoxford-iiit-petimages annotationstrimapsz;Dataset not found. You can use download=True to download itz.txtr CatDogc S s* g | ]\}}d dd |dD qS ) c s s | ]}| V qd S N)title)r partr! r! r" S s z4OxfordIIITPet.__init__..._)joinr )r raw_clsr. r! r! r" r# R s c S s$ h | ]\}}| d dd |fqS )r. r r )rsplit)r image_idlabelr! r! r" U s $ z)OxfordIIITPet.__init__..c S s | d S )Nr r! )image_id_and_labelr! r! r" V s z(OxfordIIITPet.__init__..)keyc g | ]
} j | d qS )z.jpg)_images_folderr r2 r r! r" r# \ c r8 )z.png) _segs_folderr: r r! r" r# ] r; )#r _split
isinstancestr
_target_typessuper__init__pathlibPathr _base_folderr9 _anns_folderr< _download
_check_exists RuntimeError_labels _bin_labelsopenstripr appendint bin_classessortedzipclassesdictrangelenbin_class_to_idx class_to_idx_images_segs)r r r r r r r r image_idsfileliner2 r3 bin_labelr. __class__r r" rB * sJ
zOxfordIIITPet.__init__returnc C s
t | jS r* )rV rY r r! r! r" __len___ s
zOxfordIIITPet.__len__idxc C s t | j| d}g }| jD ]'}|dkr || j| q|dkr,|| j| q|t | j| q|s=d }nt |dkrH|d }nt
|}| j rW| ||\}}||fS )NRGBr r r r ) r rL rY convertr@ rN rJ rK rZ rV tupler )r rc imagetargetr r! r! r" __getitem__b s
zOxfordIIITPet.__getitem__c C s4 | j | jfD ]}tj|rtj|s dS qdS )NFT)r9 rF ospathexistsisdir)r folderr! r! r" rH z s
zOxfordIIITPet._check_existsc C s4 | rd S | jD ]\}}t|t| j|d q d S )N)
download_rootmd5)rH
_RESOURCESr
r? rE )r urlrp r! r! r" rG s
zOxfordIIITPet._download)r r NNNF)ra N)__name__
__module__ __qualname____doc__rq r r r? rC rD r r r boolrB rO rb r r ri rH rG
__classcell__r! r! r_ r" r
s: 5r
)rj os.pathrC typingr r r r r r PILr utilsr
r visionr r
r! r! r! r" s