,}h
~ d dl Z d dlZd dlmZmZmZmZmZmZ d dl m
Z
d dl m Z ddl
mZmZ ddlmZ G d d e Zy)
N)AnyCallableListOptionalTupleUnion)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc
e Zd ZdZdZdZ ddeeej f dede
e de
e d e d
df fd
Z
d
efd Zd
ed
eeef fdZd
e fdZddZd
efdZ xZS )CLEVRClassificationa `CLEVR `_ classification dataset.
The number of objects in a scene are used as label.
Args:
root (str or ``pathlib.Path``): Root directory of dataset where directory ``root/clevr`` exists or will be saved to if download is
set to True.
split (string, optional): The dataset split, supports ``"train"`` (default), ``"val"``, or ``"test"``.
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 them 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.
z3https://dl.fbaipublicfiles.com/clevr/CLEVR_v1.0.zip b11922020e72d0cd9154779b2d3d07d2Nrootsplit transformtarget_transformdownloadreturnc t |dd | _ t |
||| t j
| j dz | _ | j t j
t | j j j z | _ |r| j | j s t d t! | j j# d| j j% d | _ | | j dk7 rt) | j d z d
| j d z 5 }t+ j, | }d d d d D ci c] }|d t/ |d
} }| j& D
cg c] }
| |
j0 c}
| _ y d gt/ | j& z | _ y # 1 sw Y wxY wc c}w c c}
w )Nr )trainvaltest)r r clevrzHDataset not found or corrupted. You can use download=True to download itimages*r scenesCLEVR_z _scenes.jsonimage_filenameobjects)r
_splitsuper__init__pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonloadlenname_labels) selfr r r r r filecontentscene num_objects
image_file __class__s Y/var/www/html/test/engine/venv/lib/python3.12/site-packages/torchvision/datasets/clevr.pyr% z CLEVRClassification.__init__ s %UG5MN
EUV #LL 3g = - - Xdii=P=U=U0V0[0[ [ NN !!# ij j "4#4#4#=#=h #T#Y#YZ]#^ _ ;;& d''(2vdkk],5WWX
*\` ))D/
*W^_gWhie5!12Ci8H4IIiKiKOK\K\]ZK
8]DL 6C(9(9$::DL
*
*i]s G*G G"Gc , t | j S N)r7 r3 r: s rA __len__zCLEVRClassification.__len__= s 4$$%% idxc | j | }| j | }t j | j d }| j
r| j | }| j r| j
| }||fS )NRGB)r3 r9 r
r4 convertr r )r: rG r? labelimages rA __getitem__z CLEVRClassification.__getitem__@ sq &&s+
S!
:&..u5 >>NN5)E ))%0Ee|rF c n | j j xr | j j S rC )r, existsis_dirrD s rA r. z!CLEVRClassification._check_existsN s+ '')Hd.?.?.F.F.HHrF c | j ry t | j t | j | j
y )N)md5)r. r r) strr( _MD5rD s rA r- z CLEVRClassification._downloadQ s0 $TYYD4E4E0FDIIVrF c d| j S )Nzsplit=)r# rD s rA
extra_reprz CLEVRClassification.extra_reprW s }%%rF )r NNF)r N)__name__
__module__ __qualname____doc__r) rT r rS r& r' r r boolr% intrE r r rM r. r- rV
__classcell__)r@ s @rA r r s
AD -D
(,/3
;C %& ; ; H% ;
#8, ;
;
;<& & s uS#X It IW &C &rF r )r5 r& typingr r r r r r urllib.parser PILr
utilsr r
visionr r rF rA rd s+ > > ! ? !L&- L&rF