kh ~ d dl Z d dlZd dlmZmZmZmZmZmZ d dl m
Z
ddl m Z ddl
mZmZ ddlmZ G d d e Zy)
N)AnyCallableListOptionalTupleUnion)urlparse )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc e Zd ZdZdZdZddddefdeee j f ded e e d
e e d e
d e eee j f gef d
df fd
Zd
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 or torch.Tensor, depends on the given loader,
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.
loader (callable, optional): A function to load an image given its path.
By default, it uses PIL as its image loader, but users could also pass in
``torchvision.io.decode_image`` for decoding image data into tensors directly.
z3https://dl.fbaipublicfiles.com/clevr/CLEVR_v1.0.zip b11922020e72d0cd9154779b2d3d07d2trainNFrootsplit transformtarget_transformdownloadloaderreturnc 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 t1 | d
}
} | j( D cg c] } |
| j2 c} | _ y d gt1 | j( z | _ y # 1 sw Y wxY wc c} w c c} w )Nr )r valtest)r r clevrzHDataset not found or corrupted. You can use download=True to download itimages*r scenesCLEVR_z _scenes.jsonimage_filenameobjects)r
_splitsuper__init__r pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonloadlenname_labels)
selfr r r r r r filecontentscene num_objects
image_file __class__s
V/var/www/teggl/fontify/venv/lib/python3.12/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/
*W^_gWhie5!12Ci8H4IIiKiKOK\K\]ZK
8]DL 6C(9(9$::DL
*
*i]s G1G$G)G!c , t | j S N)r8 r4 r; s rB __len__zCLEVRClassification.__len__B s 4$$%% idxc | j | }| j | }| j | }| j r| j | }| j r| j | }||fS rD )r4 r: r r r )r; rH r@ labelimages rB __getitem__z CLEVRClassification.__getitem__E sf &&s+
S! J' >>NN5)E ))%0Ee|rG c n | j j xr | j j S rD )r- existsis_dirrE s rB r/ z!CLEVRClassification._check_existsS s+ '')Hd.?.?.F.F.HHrG c | j ry t | j t | j | j
y )N)md5)r/ r r* strr) _MD5rE s rB r. z CLEVRClassification._downloadV s0 $TYYD4E4E0FDIIVrG c d| j S )Nzsplit=)r$ rE s rB
extra_reprz CLEVRClassification.extra_repr\ s }%%rG )r N)__name__
__module__ __qualname____doc__r* rS r r rR r' r( r r boolr r& intrF r rL r/ r. rU
__classcell__)rA s @rB r r s $ AD -D
(,/3 rc s+ > > ! " ? !Q&- Q&rG