a
h @ sp d dl Z d dlZd dlmZmZmZmZ d dlmZ ddl m
Z
ddl m Z m
Z
ddlmZ G dd d eZdS )
N)AnyCallableOptionalUnion)urlparse )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc
s e Zd ZdZdZdZddddefeee j
f ee e e e e
e eee j
f gef dd fdd
Zed
d d Zeeeef d
ddZe
d
ddZdd
ddZed
ddZ ZS )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.zipZ b11922020e72d0cd9154779b2d3d07d2trainNF)rootsplit transformtarget_transformdownloadloaderreturnc s" t |dd| _t j|||d || _t| jd | _| jtt | j
j j | _
|r`| | sptdt| j
d| jd| _| | jdkr t| j
d d
| j d }t|}W d n1 s0 Y d d
|d D fdd| jD | _nd gt| j | _d S )Nr )r
valtest)r r ZclevrzHDataset not found or corrupted. You can use download=True to download itZimages*r ZscenesZCLEVR_z _scenes.jsonc S s i | ]}|d t |d qS )Zimage_filenameZobjects)len).0Zscene r H/var/www/auris/lib/python3.9/site-packages/torchvision/datasets/clevr.py
= z0CLEVRClassification.__init__..c s g | ]} |j qS r )name)r
image_file num_objectsr r
> r z0CLEVRClassification.__init__..)r
_splitsuper__init__r pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonload_labelsr ) selfr r r r r r filecontent __class__r r r% " s" (z CLEVRClassification.__init__)r c C s
t | jS N)r r3 r8 r r r __len__B s zCLEVRClassification.__len__)idxr c C sF | j | }| j| }| |}| jr.| |}| jr>| |}||fS r= )r3 r7 r r r )r8 r@ r labelimager r r __getitem__E s
z CLEVRClassification.__getitem__c C s | j o| j S r= )r, existsis_dirr> r r r r. S s z!CLEVRClassification._check_existsc C s( | r d S t| jt| j| jd d S )N)md5)r. r r) strr( _MD5r> r r r r- V s z CLEVRClassification._downloadc C s d| j S )Nzsplit=)r# r> r r r
extra_repr\ s z CLEVRClassification.extra_repr)__name__
__module__ __qualname____doc__r) rH r r rG r& r' r r boolr r% intr? tuplerC r. r- rI
__classcell__r r r; r r s, r )r5 r& typingr r r r urllib.parser folderr utilsr r
Zvisionr r r r r r s