Di v d dl Z d dlZd dlmZmZmZmZ d dlmZ ddl m
Z
ddl m Z m
Z
ddlmZ G d d e Zy)
N)AnyCallableOptionalUnion)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
\/var/www/pod-logistic/pod-ai/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)r6 r2 r9 s r@ __len__zCLEVRClassification.__len__B s 4$$%% idxc | j | }| j | }| j | }| j r| j | }| j r| j | }||fS rB )r2 r8 r r r )r9 rF r> labelimages r@ __getitem__z CLEVRClassification.__getitem__E sf &&s+
S! J' >>NN5)E ))%0Ee|rE c n | j j xr | j j S rB )r+ existsis_dirrC s r@ r- z!CLEVRClassification._check_existsS 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._downloadV s0 $TYYD4E4E0FDIIVrE c d| j S )Nzsplit=)r" rC s r@
extra_reprz CLEVRClassification.extra_repr\ s }%%rE )r N)__name__
__module__ __qualname____doc__r( rQ r r rP r% r& r r boolr r$ intrD tuplerJ r- r, rS
__classcell__)r? s @r@ r r s $ AD -D
(,/3 rb s+ 1 1 ! " ? !Q&- Q&rE