o
V h @ 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.zip 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 clevrzHDataset not found or corrupted. You can use download=True to download itimages*r scenesCLEVR_z _scenes.jsonc S s i | ] }|d t |d qS )image_filenameobjects)len).0scene r$ N/var/www/vscode/kcb/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! r< rA r$ r$ r% __len__B s
zCLEVRClassification.__len__idxc C sF | j | }| j| }| |}| jr| |}| jr | |}||fS rF )r< r@ r r r )rA rI r( labelimager$ r$ r% __getitem__E s
z CLEVRClassification.__getitem__c C s | j o | j S rF )r5 existsis_dirrG r$ r$ r% r7 S s z!CLEVRClassification._check_existsc C s( | rd S t| jt| j| jd d S )N)md5)r7 r r2 strr1 _MD5rG r$ r$ r% r6 V s z CLEVRClassification._downloadc C s d| j S )Nzsplit=)r, rG r$ r$ r%
extra_repr\ s z CLEVRClassification.extra_repr)r N)__name__
__module__ __qualname____doc__r2 rQ r
r rP r/ r0 r r boolr r. intrH r rL r7 r6 rR
__classcell__r$ r$ rD r% r s:
r )r> r/ typingr r r r r r urllib.parser folderr
utilsr r visionr
r r$ r$ r$ r% s