o
)iX
@ st d dl Z d dlZd dlmZmZmZmZmZ d dlm Z d dl
m Z ddl m
Z
mZ ddlmZ G dd d eZdS )
N)AnyCallableListOptionalTuple)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc
s e Zd ZdZdZdZ ddeded ee d
ee d e d df fd
d
Z
d e fddZ de d e
eef fddZd e fddZdddZd efddZ ZS ) CLEVRClassificationap `CLEVR `_ classification dataset.
The number of objects in a scene are used as label.
Args:
root (string): 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 an 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.zipZ 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 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 i/var/www/html/eduruby.in/lip-sync/lip-sync-env/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 r3 r8 r r r __len__= s
zCLEVRClassification.__len__idxc C sL | j | }| j| }t|d}| jr| |}| jr"| |}||fS )NRGB)r3 r7 r r4 convertr r )r8 r@ r labelimager r r __getitem__@ s
z CLEVRClassification.__getitem__c C s | j o | j S r= )r, existsis_dirr> r r r r. N s z!CLEVRClassification._check_existsc C s( | rd S t| jt| j| jd d S )N)md5)r. r
r) strr( _MD5r> r r r r- Q s z CLEVRClassification._downloadc C s d| j S )Nzsplit=)r# r> r r r
extra_reprW s z CLEVRClassification.extra_repr)r NNF)r N)__name__
__module__ __qualname____doc__r) rJ rI r r boolr% intr? r r rE r. r- rK
__classcell__r r r; r r
s4
r
)r5 r& typingr r r r r urllib.parser ZPILr utilsr
r Zvisionr r
r r r r s