Ph: z d dl Z d dlZ d dlZd dlmZmZmZmZmZm Z d dl
m Z ddl m
Z
mZ ddlmZ G d de Zy) N)AnyCallableOptionalSequenceTupleUnion)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZ ddededeee ef de e
de e
d e e
d
e f fd
Z d e
fd
Zde
d eeef fdZd e fdZddZ xZS )
OxfordIIITPeta `Oxford-IIIT Pet Dataset `_.
Args:
root (string): 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.
- ``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.gz 5c4f3ee8e5d25df40f4fd59a7f44e54c)zBhttps://www.robots.ox.ac.uk/~vgg/data/pets/data/annotations.tar.gz 95a8c909bbe2e81eed6a22bccdf3f68f)category segmentationrootsplit target_types
transforms transformtarget_transformdownloadc
t |dd | _ t |t r|g}|D cg c] }t |d| j c}| _ t | |||| t j | j dz | _ | j dz | _ | j dz | _
| j dz | _ |r| j | j! s t# d g } g | _ t' | j | j d
z 5 }
|
D ][ } | j) j+ ^} }
}| j- | | j$ j- t/ |
d z
] d d d t1 t3 | | j$ D
ch c] \ } }
| j5 d d d
|
f c}
} d D cg c], \ }}dj7 d |j+ d D . c}}| _ t; t3 | j8 t= t? | j8 | _ | D cg c] } | j | dz c} | _! | D cg c] } | j | dz c} | _" y c c}w # 1 sw Y !xY wc c}
} w c c}}w c c} w c c} w )Nr )trainvaltestr )r r r zoxford-iiit-petimages annotationstrimapsz;Dataset not found. You can use download=True to download itz.txtr
_r c | d S )Nr
)image_id_and_labels oC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/oxford_iiit_pet.pyz(OxfordIIITPet.__init__..P s
/A!/D )key c 3 < K | ] }|j y wN)title).0parts r% z)OxfordIIITPet.__init__..M s A.@dTZZ\.@s z.jpgz.png)#r _split
isinstancestr_VALID_TARGET_TYPES
_target_typessuper__init__pathlibPathr _base_folder_images_folder _anns_folder _segs_folder _download
_check_exists RuntimeError_labelsopenstripr appendintsortedziprsplitjoinclassesdictrangelen class_to_idx_images_segs)selfr r r r r r r target_type image_idsfilelineimage_idlabelr! raw_cls __class__s r% r6 zOxfordIIITPet.__init__' s %UG5IJ lC ( (>Leq
eqVaN;8P8P Qeq
* \lm #LL 36G G " / /( : - -
= - - 9 NN !!# \] ]
$##
T&::
;t &*jjl&8&8&:#%! * ##CJN3 < %LOPY[_[g[gLhiLh 5(//#q)!,e4LhiD
HHAgmmC.@A A
!T\\5T\\9J3K!L MPYZPYH++
$.??PYZ LUVId''XJd*;;IV
E
$ <
; j
[Vs* J "A!J#*!J0
1J6J<<K#J-returnc , t | j S r+ )rL rN )rP s r% __len__zOxfordIIITPet.__len__X s 4<< r' idxc t j | j | j d }g }| j D ]W }|dk( r |j | j | '|j t j | j | Y |sd }n t | dk( r|d }n t | }| j r| j || \ }}||fS )NRGBr r
r ) r rA rN convertr4 rC r@ rO rL tupler )rP r\ imagetargetrQ s r% __getitem__zOxfordIIITPet.__getitem__[ s
4<<,-55e< --Kj(
dll3 /0
ejjC 9: . F
[A
AYF6]F ?? OOE6 :ME6f} r' c | j | j fD ]B }t j j | r t j j | rB y y)NFT)r: r; ospathexistsisdir)rP folders r% r> zOxfordIIITPet._check_existsq sD **D,=,=>FGGNN6*rww}}V/D ? r' c | j ry | j D ]& \ }}t |t | j | ( y )N)
download_rootmd5)r>
_RESOURCESr r2 r9 )rP urlrl s r% r= zOxfordIIITPet._downloadx s; HC (C@Q@Q r=
__classcell__)rX s @r% r r s (J 7
2<)-(,/3 /W/W /W HSM3 ./ /W
X& /W H%
/W #8,/W /Wb! ! s uS#X ,t ]r' r )re os.pathr7 typingr r r r r r PILr utilsr r visionr
r r# r' r% rz s- B B ? !q]M q]r'