h v S SK r S SKrS SKJrJrJrJr S SKJr SSK J
r
SSK J r J
r
SSKJr " S S \5 rg)
N)AnyCallableOptionalUnion)urlparse )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc ^ \ rS rSrSrSrSrSSSS\4S\\ \
R 4 S \ S
\ \
S \ \
S \S
\
\\ \
R 4 /\4 SS4U 4S jjjrS\4S jrS\S\\\4 4S jrS\4S jrSS jrS\ 4S jrSrU =r$ )CLEVRClassification ax `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 > [ USS5 U l [ T U ]
XUS9 X`l [
R " U R 5 S- U l U R [
R " [ U R 5 R 5 R - U l
U( a U R 5 U R 5 ( d [! S5 e[# U R R% SU R 5 R' S5 5 U l U U R S:w a [+ U R S - S
U R S 3- 5 n[, R. " U5 nS S S 5 WS V s0 s H oS [1 U S
5 _M n
n U R( V s/ s H oU R2 PM sn U l g S /[1 U R( 5 - U l g ! , ( d f N= fs sn f s sn f )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
R/var/www/auris/envauris/lib/python3.13/site-packages/torchvision/datasets/clevr.pyr% 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^_gWhiWhe!12Ci8H4IIWhKiKOK\K\]K\Z
8K\]DL 6C(9(9$::DL YXi]s G%:G6%G;%
G3c , [ U R 5 $ N)r7 r3 r: s rA __len__CLEVRClassification.__len__B s 4$$%% idxc U R U nU R U nU R U5 nU R ( a U R U5 nU R ( a U R U5 nXC4$ rD )r3 r9 r r r )r: rI r? labelimages rA __getitem__ CLEVRClassification.__getitem__E sd &&s+
S! J' >>NN5)E ))%0E|rH c x U R R 5 =( a U R R 5 $ rD )r, existsis_dirrE s rA r. !CLEVRClassification._check_existsS s+ '')Hd.?.?.F.F.HHrH c U R 5 ( a g [ U R [ U R 5 U R
S9 g )N)md5)r. r
r) strr( _MD5rE s rA r- CLEVRClassification._downloadV s3 $TYYD4E4E0FDIIVrH c SU R 3$ )Nzsplit=)r# rE s rA
extra_repr CLEVRClassification.extra_repr\ s }%%rH )r( r, r3 r9 r# r )r N)__name__
__module__ __qualname____firstlineno____doc__r) rV r r rU r& r' r r boolr r% intrF tuplerM r. r- rY __static_attributes__
__classcell__)r@ s @rA r r s $ AD -D
(,/3 rk s+ 1 1 ! " ? !Q&- Q&rH