U
ʹjX
@ 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 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_filenameobjects)len).0Zscene r Y/var/www/html/bwave/python/venv/lib/python3.8/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__)r c C s
t | jS N)r r3 r8 r r r __len__= s zCLEVRClassification.__len__)idxr c C sL | j | }| j| }t|d}| jr4| |}| jrD| |}||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( | 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) rJ rI r r boolr% intr? r r rE r. r- rK
__classcell__r r r; r r
s( r
)r5 r& typingr r r r r urllib.parser ZPILr utilsr
r Zvisionr r
r r r r s