o
;)i @ sx d dl Z d dlZd dlmZmZmZmZmZmZ d dl m
Z
ddl m Z ddl
mZmZ ddlmZ G dd d eZdS )
N)AnyCallableListOptionalTupleUnion)urlparse )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc s e Zd ZdZdZdZddddefdeee j
f ded e e d
e e d e
d e eee j
f gef d
df fdd
Zd
efddZded
eeef fddZd
e
fddZdddZd
efddZ 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 b11922020e72d0cd9154779b2d3d07d2trainNFrootsplit transformtarget_transformdownloadloaderreturnc s" t |dd| _t j|||d || _t| jd | _| jtt | j
j j | _
|r0| | s8tdt| j
d| jd| _| | jdkrt| j
d d
| j d
}t|}W d n1 skw Y d d
|d D fdd| jD | _d S d 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_filenameobjects)len).0Zscene r U/home/app/PyTorch/.pytorch/lib/python3.10/site-packages/torchvision/datasets/clevr.py
= s z0CLEVRClassification.__init__..c s g | ]} |j qS r )name)r
image_file num_objectsr r
> s 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__c C s
t | jS N)r r5 r: r r r __len__B s
zCLEVRClassification.__len__idxc C sF | j | }| j| }| |}| jr| |}| jr | |}||fS r? )r5 r9 r r r )r: rB r! labelimager r r __getitem__E s
z CLEVRClassification.__getitem__c C s | j o | j S r? )r. existsis_dirr@ r r r r0 S s z!CLEVRClassification._check_existsc C s( | rd S t| jt| j| jd d S )N)md5)r0 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)r N)__name__
__module__ __qualname____doc__r+ rJ r
r rI r( r) r r boolr r' intrA r rE r0 r/ rK
__classcell__r r r= r r s:
r )r7 r( typingr r r r r r urllib.parser folderr
utilsr r Zvisionr
r r r r r s