j
d dl Z d dlZd dlmZmZmZmZmZmZ d dl m
Z
d dl m Z ddl
mZmZ ddlmZ G d d e ZdS )
N)AnyCallableListOptionalTupleUnion)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc
e Zd ZdZdZdZ ddeeej f ded e
e d
e
e d e d df fd
Z
d efdZded eeef fdZd e fdZddZd efdZ xZS )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 t |dd | _ t ||| t j | j dz | _ | j t j t | j j
j z | _ |r|
| st d t! | j d| j d | _ | | j dk r{t) | j d z d
| j d z 5 }t+ j | }d d d n # 1 swxY w Y d |d D fd
| j D | _ d S d gt1 | j z | _ 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 F i | ] }|d t |d S )image_filenameobjects)len).0scenes KC:\Projects\Historical\venv\Lib\site-packages\torchvision/datasets/clevr.py
z0CLEVRClassification.__init__..8 s- iiie5!12Ci8H4I4Iiii c * g | ]}|j S )name)r%
image_file num_objectss r'
z0CLEVRClassification.__init__..9 s ]]]ZK
8]]]r) )r
_splitsuper__init__pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonload_labelsr$ )
selfr r r r r filecontentr. __class__s
@r' r2 z CLEVRClassification.__init__ s %UG5MNN
EUVVV #L 3 3g = - Xdi=P=P=U0V0V0[ [ NN !!## k ijj j "4#4#=#=h #T#T#Y#YZ]#^#^ _ _ ) ;& d'(25Wdk5W5W5WWXX
*\` )D//
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*iiW^_gWhiiiK]]]]4K\]]]DLLL 6C(9$:$::DLLLs E**E.1E.c * t | j S N)r$ r@ rE s r' __len__zCLEVRClassification.__len__= s 4$%%%r) idxc | j | }| j | }t j | d }| j r| | }| j r| | }||fS )NRGB)r@ rD r
rA convertr r )rE rM r- labelimages r' __getitem__z CLEVRClassification.__getitem__@ s{ &s+
S!
:&&..u55 > *NN5))E 1))%00Ee|r) c f | j o| j S rJ )r9 existsis_dirrK s r' r; z!CLEVRClassification._check_existsN s+ ''))Hd.?.F.F.H.HHr) c | rd S t | j t | j | j d S )N)md5)r; r r6 strr5 _MD5rK s r' r: z CLEVRClassification._downloadQ sE F$TYD4E0F0FDIVVVVVVr) c d| j S )Nzsplit=)r0 rK s r'
extra_reprz CLEVRClassification.extra_reprW s % %%%r) )r NNF)r N)__name__
__module__ __qualname____doc__r6 rZ r rY r3 r4 r r boolr2 intrL r r rS r; r: r\
__classcell__)rH s @r' r r sO
AD -D
(,/3
; ;C %& ; ; H% ;
#8, ;
;
; ; ; ; ; ;<& & & & & s uS#X It I I I IW W W W &C & & & & & & & &r) r )rB r3 typingr r r r r r urllib.parser PILr
utilsr r
visionr r r+ r) r' ri s > > > > > > > > > > > > > > > > ! ! ! ! ! ! ? ? ? ? ? ? ? ? ! ! ! ! ! !L& L& L& L& L&- L& L& L& L& L&r)