o
0i
@ sx d dl Z d dlZd dlmZmZmZmZmZmZ d dl m
Z
d dl m Z ddl
mZmZ ddlmZ G dd d eZdS )
N)AnyCallableListOptionalTupleUnion)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc
s e Zd ZdZdZdZ ddeeej f ded e
e d
e
e d e d df fd
d
Z
d 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 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 b11922020e72d0cd9154779b2d3d07d2trainNFrootsplit transformtarget_transformdownloadreturnc s t |dd| _t j|||d t| jd | _| jtt| j j
j | _ |r-|
| s5tdt| j d| jd| _| | jdkrt| j d d
| j d
}t|}W d n1 shw 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# X/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torchvision/datasets/clevr.py
8 s z0CLEVRClassification.__init__..c s g | ]} |j qS r# )name)r!
image_file num_objectsr# r$
9 s z0CLEVRClassification.__init__..)r _splitsuper__init__pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonload_labelsr )selfr r r r r filecontent __class__r( r$ r- s"
z CLEVRClassification.__init__c C s
t | jS N)r r; r@ r# r# r$ __len__= s
zCLEVRClassification.__len__idxc C sL | j | }| j| }t|d}| jr| |}| jr"| |}||fS )NRGB)r; r? r r< convertr r )r@ rH r' labelimager# r# r$ __getitem__@ s
z CLEVRClassification.__getitem__c C s | j o | j S rE )r4 existsis_dirrF r# r# r$ r6 N s z!CLEVRClassification._check_existsc C s( | rd S t| jt| j| jd d S )N)md5)r6 r r1 strr0 _MD5rF r# r# r$ r5 Q s z CLEVRClassification._downloadc C s d| j S )Nzsplit=)r+ rF r# r# r$
extra_reprW s z CLEVRClassification.extra_repr)r NNF)r N)__name__
__module__ __qualname____doc__r1 rR r rQ r. r/ r r boolr- intrG r r rM r6 r5 rS
__classcell__r# r# rC r$ r s4
r )r= r. typingr r r r r r urllib.parser PILr utilsr r visionr
r r# r# r# r$ s