U
(dX
@ st d dl Z d dlZd dlmZmZmZmZmZ d dlm Z d dl
m Z ddl m
Z
mZ ddlmZ G dd d eZdS )
N)AnyCallableOptionalTupleList)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc s e Zd ZdZdZdZdeeee ee e dd fdd
Z
e d
d d Z e e
eef d
ddZe d
ddZdd
ddZed
ddZ 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 b11922020e72d0cd9154779b2d3d07d2trainNF)rootsplit transformtarget_transformdownloadreturnc s t |dd| _t j|||d t| jd | _| jtt| j j
j | _ |rZ|
| sjtdt| j d| jd| _| | jdkrt| j d d
| j d }t|}W 5 Q R X 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 >/tmp/pip-unpacked-wheel-vx7f76es/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__)r c C s
t | jS N)r r2 r7 r r r __len__= s zCLEVRClassification.__len__)idxr c C sL | j | }| j| }t|d}| jr4| |}| jrD| |}||fS )NZRGB)r2 r6 r r3 convertr r )r7 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( | r d 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)__name__
__module__ __qualname____doc__r( rH rG r r boolr$ intr> r r rC r- r, rI
__classcell__r r r: r r
s( r
)r4 r% typingr r r r r urllib.parser ZPILr utilsr
r Zvisionr r
r r r r s