^j@ ~ d dl Z d dlZ d dlZd dlmZ d dlmZmZmZm Z d dl
m Z ddl m
Z
mZ ddlmZ G d d e Zy)
N)Sequence)AnyCallableOptionalUnion)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZ ddeeej f dedee
e ef de e de e d e e d
e
f fd
Zd efd
Zded eeef fdZd e
fdZddZ xZS )
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 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 | _ g | _ t) | j | j d
z 5 }
|
D ] } | j+ j- \ } }
}}| j/ | | j$ j/ t1 |
d z
| j& j/ t1 | d z
d d d d d
g| _ t5 t7 | | j$ D
ch c] \ } }
| j9 dd d |
f c}
} d D cg c], \ }}dj; d |j- d D . c}}| _ t? t7 | j2 tA tC | j2 | _" t? t7 | j< tA tC | j< | _# | D cg c] } | j | dz c} | _$ | D cg c] } | j | dz c} | _% y c c}w # 1 sw Y jxY 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 CatDog_r c | d S )Nr )image_id_and_labels o/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torchvision/datasets/oxford_iiit_pet.pyz(OxfordIIITPet.__init__..W s /A!/D )key c 3 < K | ] }|j y wN)title).0parts r' z)OxfordIIITPet.__init__..T s AdTZZ\As 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_labels _bin_labelsopenstripr appendint bin_classessortedziprsplitjoinclassesdictrangelenbin_class_to_idx class_to_idx_images_segs)selfr r r r r r r target_type image_idsfilelineimage_idlabel bin_labelr# raw_cls __class__s r' r8 zOxfordIIITPet.__init__+ s %UG5IJ lC ( (>Leq
VaN;8P8P Q
* \lm #LL 36G G " / /( : - -
= - - 9 NN !!# \] ]
$##
T&::
; %LOPY[_[g[gLhi 5(//#q)!,e4iD
HHAgmmC.@A A
!%S)9)95TEUEUAV;W%X Y T\\5T\\9J3K!L MPYZH++
$.??Z LUVd''XJd*;;V
M
& < < j
[Vs* L)B
L #!L)
1L/L55L: L&returnc , t | j S r- )rP rS )rU s r' __len__zOxfordIIITPet.__len__` s 4<< r) idxc t j | j | j d }g }| j D ]{ }|dk( r |j | j | '|dk( r |j | j | K|j t j | j | } |sd }n t | dk( r|d }n t | }| j r| j || \ }}||fS )NRGBr r r r ) r rD rS convertr6 rF rB rC rT rP tupler )rU rb imagetargetrV s r' __getitem__zOxfordIIITPet.__getitem__c s
4<<,-55e< -- ;Kj(
dll3 /0 11
d . .s 34
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)rU folders r' r@ zOxfordIIITPet._check_exists{ sI **D,=,=> FGGNN6*rww}}V/D r) c | j ry | j D ]& \ }}t |t | j | ( y )N)
download_rootmd5)r@
_RESOURCESr
r4 r; )rU urlrr s r' r? zOxfordIIITPet._download sB ]HC (C@Q@Qr s0 $ 1 1 ? !z]M z]r)