Ph
z d dl Z d dlZd dlmZmZmZmZmZ d dlm Z d dl
m Z ddl m
Z
mZ ddlmZ G d d e Zy)
N)AnyCallableListOptionalTuple)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc
e Zd ZdZdZdZ ddededee dee d e d
df fd
Z
d
e fd Z d
e d
e
eef fdZd
e fdZddZd
efdZ xZS )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.zip b11922020e72d0cd9154779b2d3d07d2Nrootsplit transformtarget_transformdownloadreturnc t |dd | _ t |
||| t j
| j dz | _ | j t j
t | j j j z | _ |r| j | j s t d t! | j j# d| j j% d | _ | | j dk7 rt) | j d z d
| j d z 5 }t+ j, | }d d d d D ci c] }|d t/ |d
} }| j& D
cg c] }
| |
j0 c}
| _ y d gt/ | j& z | _ y # 1 sw Y wxY wc c}w c c}
w )Nr )trainvaltest)r r clevrzHDataset not found or corrupted. You can use download=True to download itimages*r scenesCLEVR_z _scenes.jsonimage_filenameobjects)r _splitsuper__init__pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonloadlenname_labels) selfr r r r r filecontentscene num_objects
image_file __class__s eC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/clevr.pyr$ z CLEVRClassification.__init__ s %UG5MN
EUV #LL 3g = - - Xdii=P=U=U0V0[0[ [ NN !!# ij j "4#4#4#=#=h #T#Y#YZ]#^ _ ;;& d''(2vdkk],5WWX\` ))D/ YW^_gWhiWhe5!12Ci8H4IIWhKiKOK\K\]K\ZK
8K\]DL 6C(9(9$::DL YXi]s G*G G"Gc , t | j S N)r6 r2 r9 s r@ __len__zCLEVRClassification.__len__= s 4$$%% idxc | j | }| j | }t j | j d }| j
r| j | }| j r| j
| }||fS )NRGB)r2 r8 r r3 convertr r )r9 rF r> labelimages r@ __getitem__z CLEVRClassification.__getitem__@ sq &&s+
S!
:&..u5 >>NN5)E ))%0Ee|rE c n | j j xr | j j S rB )r+ existsis_dirrC s r@ r- z!CLEVRClassification._check_existsN s+ '')Hd.?.?.F.F.HHrE c | j ry t | j t | j | j
y )N)md5)r- r r( strr' _MD5rC s r@ r, z CLEVRClassification._downloadQ s0 $TYYD4E4E0FDIIVrE c d| j S )Nzsplit=)r" rC s r@
extra_reprz CLEVRClassification.extra_reprW s }%%rE )r NNF)r N)__name__
__module__ __qualname____doc__r( rS rR r r boolr$ intrD r r rL r- r, rU
__classcell__)r? s @r@ r r s
AD -D
(,/3
; ; ; H% ;
#8, ;
;
;<& & s uS#X It IW &C &rE r )r4 r% typingr r r r r urllib.parser PILr utilsr r visionr
r rE r@ rc s+ 7 7 ! ? !L&- L&rE