QЦi
~ S SK r S SKrS SKJrJrJrJrJrJr S SK J
r
S SK J r SSK
JrJr SSKJr " S S \5 rg)
N)AnyCallableListOptionalTupleUnion)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc
^ \ rS rSrSrSrSr SS\\\ R 4 S\S\ \ S \ \ S
\
S S4 U 4S jjjrS \4S
jrS\S \\\4 4S jrS \
4S jrSS jrS \4S jrSrU =r$ )CLEVRClassification aT `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 b11922020e72d0cd9154779b2d3d07d2Nrootsplit transformtarget_transformdownloadreturnc > [ USS5 U l [ T U ]
XUS9 [ 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 Vs0 s H oS [/ US
5 _M n nU R& V
s/ s H oU
R0 PM sn
U l g S /[/ U R& 5 - U l g ! , ( d f N= fs snf s sn
f )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 Y/var/www/html/ai-image-ml/venv/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 4G0 G5
G-c , [ U R 5 $ N)r8 r4 r; s rB __len__CLEVRClassification.__len__= s 4$$%% idxc U R U nU R U n[ R " U5 R S5 nU R
( a U R U5 nU R ( a U R
U5 nXC4$ )NRGB)r4 r: r
r5 convertr r )r; rJ r@ labelimages rB __getitem__ CLEVRClassification.__getitem__@ so &&s+
S!
:&..u5 >>NN5)E ))%0E|rI c x U R R 5 =( a U R R 5 $ rE )r- existsis_dirrF s rB r/ !CLEVRClassification._check_existsN s+ '')Hd.?.?.F.F.HHrI c U R 5 ( a g [ U R [ U R 5 U R
S9 g )N)md5)r/ r r* strr) _MD5rF s rB r. CLEVRClassification._downloadQ s3 $TYYD4E4E0FDIIVrI c SU R 3$ )Nzsplit=)r$ rF s rB
extra_repr CLEVRClassification.extra_reprW s }%%rI )r) r- r4 r: r$ )r NNF)r N)__name__
__module__ __qualname____firstlineno____doc__r* rY r rX r' r( r r boolr& intrG r r rP r/ r. r\ __static_attributes__
__classcell__)rA s @rB r r s
AD -D
(,/3
;C %& ; ; H% ;
#8, ;
;
; ;<& & s uS#X It IW &C & &rI r )r6 r' typingr r r r r r urllib.parser PILr
utilsr r
visionr r rI rB rm s+ > > ! ? !L&- L&rI