Yh ~ d dl Z d dlZd dlmZmZmZmZ d dlmZ ddl m
Z
ddl m Z m
Z
ddlmZ G d d e ZdS )
N)AnyCallableOptionalUnion)urlparse )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZddddefdeee j
f ded e e d
e e d e
d e eee j
f gef d
df fd
Zd
efdZded
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 or torch.Tensor, depends on the given loader,
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.
loader (callable, optional): A function to load an image given its path.
By default, it uses PIL as its image loader, but users could also pass in
``torchvision.io.decode_image`` for decoding image data into tensors directly.
z3https://dl.fbaipublicfiles.com/clevr/CLEVR_v1.0.zip b11922020e72d0cd9154779b2d3d07d2trainNFrootsplit transformtarget_transformdownloadloaderreturnc t |dd | _ t ||| || _ t j | j dz | _ | j t j t | j
j j z | _
|r| | st! d t# | j
d| j d | _ | | j dk r{t+ | j
d z d
| j d z 5 }t- j | }d d d n # 1 swxY w Y d |d D fd
| j D | _ d S d gt3 | j z | _ d S )Nr )r valtest)r r clevrzHDataset not found or corrupted. You can use download=True to download itimages*r scenesCLEVR_z _scenes.jsonc F i | ] }|d t |d S )image_filenameobjects)len).0scenes l/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/clevr.py
z0CLEVRClassification.__init__..= s- iiie5!12Ci8H4I4Iiii c * g | ]}|j S )name)r$
image_file num_objectss r&
z0CLEVRClassification.__init__..> s ]]]ZK
8]]]r( )r _splitsuper__init__r pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonload_labelsr# ) selfr r r r r r filecontentr- __class__s @r& r1 z CLEVRClassification.__init__" s %UG5MNN
EUVVV #L 3 3g = - Xdi=P=P=U0V0V0[ [ NN !!## k ijj j "4#4#=#=h #T#T#Y#YZ]#^#^ _ _ ) ;& d'(25Wdk5W5W5WWXX
*\` )D//
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*iiW^_gWhiiiK]]]]4K\]]]DLLL 6C(9$:$::DLLLs E11E58E5c * t | j S N)r# r? rD s r& __len__zCLEVRClassification.__len__B s 4$%%%r( idxc | j | }| j | }| | }| j r| | }| j r| | }||fS rI )r? rC r r r )rD rL r, labelimages r& __getitem__z CLEVRClassification.__getitem__E so &s+
S! J'' > *NN5))E 1))%00Ee|r( c f | j o| j S rI )r8 existsis_dirrJ s r& r: z!CLEVRClassification._check_existsS s+ ''))Hd.?.F.F.H.HHr( c | rd S t | j t | j | j d S )N)md5)r: r
r5 strr4 _MD5rJ s r& r9 z CLEVRClassification._downloadV sE F$TYD4E0F0FDIVVVVVVr( c d| j S )Nzsplit=)r/ rJ s r&
extra_reprz CLEVRClassification.extra_repr\ s % %%%r( )r N)__name__
__module__ __qualname____doc__r5 rW r r rV r2 r3 r r boolr r1 intrK tuplerP r: r9 rY
__classcell__)rG s @r& r r sv $ AD -D
(,/3 rg s 1 1 1 1 1 1 1 1 1 1 1 1 ! ! ! ! ! ! " " " " " " ? ? ? ? ? ? ? ? ! ! ! ! ! !Q& Q& Q& Q& Q&- Q& Q& Q& Q& Q&r(