+
j" ^ RI t ^ RIt^ RIt^ RIHt ^ RIHt ^ RIHt ^ RI H
t
H t H t ^ RI
t
^RIHt ^RIHtHtHt RRR RR
R/tR t ! R R
]4 tRR R lltR R ltRR R lltR R R lltR R R lltR# ) N)Iterator)contextmanager)Path)AnyOptionalUnion) ImageFolder)check_integrityextract_archiveverify_str_argtrainvaldevkitzmeta.binc a a ] tR t^t oRtR V3R lV 3R llltV3R lR lt]V3R lR l4 tV3R lR lt R
t
Vt V ;t # ) ImageNeta `ImageNet `_ 2012 Classification Dataset.
.. note::
Before using this class, it is required to download ImageNet 2012 dataset from
`here `_ and
place the files ``ILSVRC2012_devkit_t12.tar.gz`` and ``ILSVRC2012_img_train.tar``
or ``ILSVRC2012_img_val.tar`` based on ``split`` in the root directory.
Args:
root (str or ``pathlib.Path``): Root directory of the ImageNet Dataset.
split (string, optional): The dataset split, supports ``train``, or ``val``.
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 the
target and transforms it.
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.
Attributes:
classes (list): List of the class name tuples.
class_to_idx (dict): Dict with items (class_name, class_index).
wnids (list): List of the WordNet IDs.
wnid_to_idx (dict): Dict with items (wordnet_id, class_index).
imgs (list): List of (image path, class_index) tuples
targets (list): The class_index value for each image in the dataset
c F < V ^8 d QhRS[ S[S[3, RS[RS[RR/# ) rootsplitkwargsreturnN)r strr r )format
__classdict__s "^/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchvision/datasets/imagenet.py __annotate__ImageNet.__annotate__4 s9
`
`U39 -
`c
`s
`W[
` c , < \ P P V4 ;qn \ VR R4 V n V P
4 \ V P 4 ^ , p\ S V `$ ! V P 3/ VB Wn V P V n V P V n V P Uu. uF qTV, NK
upV n \ V P 4 UUUu/ uF w rgV F qVbK K upppV n
R# u upi u upppi )r N)r
r )ospath
expanduserr r r parse_archivesload_meta_filesuper__init__ split_folderclasseswnids class_to_idx wnid_to_idx enumerate)
selfr r r wnid_to_classeswnididxclsscls __class__s
&&&, r r& ImageNet.__init__4 s 77--d33y#E74DE
(3A6
**5f5 \\
,,:>**E*$--*E 7@ 7N _7N)#Z^SV#XZ^S7N _ F _s 6D
(Dc < V ^8 d QhRR/# )r r N )r r s "r r r C s - - -r c \ \ P P V P \
4 4 ' g \
V P 4 \ P P V P 4 ' gS V P R 8X d \ V P 4 R# V P R8X d \ V P 4 R# R# R# )r
r N) r
r r! joinr META_FILEparse_devkit_archiveisdirr' r parse_train_archiveparse_val_archiver- s &r r# ImageNet.parse_archivesC s rww||DIIy ABB +ww}}T . .//zzW$#DII.u$!$)), % 0r c < V ^8 d QhRS[ /# r r r )r r s "r r r N s 3 3c 3r c j \ P P V P V P 4 # N)r r! r8 r r r> s &r r' ImageNet.split_folderM s ww||DIItzz22r c < V ^8 d QhRS[ /# rA rB )r r s "r r r Q s 8 8C 8r c : R P ! R/ V P B # )zSplit: {split}r6 )r __dict__r> s &r
extra_reprImageNet.extra_reprQ s &&7 77r )r* r( r r r+ r) )r
)
__name__
__module__ __qualname____firstlineno____doc__r& r# propertyr' rI __static_attributes____classdictcell__
__classcell__)r3 r s @@r r r sC 8
`
` - - 3 38 8 8r r c V ^8 d QhR\ \ \ 3, R\ \ , R\ \
\ \ 3, \ \ , 3, /# )r r filer )r r r r tupledictlist)r s "r r r U sL 3 3sDy) 3# 3%PTUXZ]U]P^`deh`iPiJj 3r c Vf \ p\ P P W4 p\ V4 ' d \
P ! VRR7 # Rp\ VP W4 4 h)NT) weights_onlyzThe meta file {} is not present in the root directory or is corrupted. This file is automatically created by the ImageNet dataset.) r9 r r! r8 r
torchload RuntimeErrorr )r rU msgs && r r$ r$ U s[ |
77<< #Dt zz$T22
J 3::d122r c b V ^8 d QhR\ \ \ 3, R\ R\ RR/# )r r rU md5r N)r r r )r s "r r r d s1 3 3%T * 3# 3C 3D 3r c \ \ P P W4 V4 ' g R p\ VP W4 4 hR# )z{The archive {} is not present in the root directory or is corrupted. You need to download it externally and place it in {}.N)r
r r! r8 r] r )r rU r` r^ s &&& r _verify_archiverb d sB 277<<3S 9 9
E 3::d122 :r c l V ^8 d QhR\ \ \ 3, R\ \ , RR/# )r r rU r Nr r r r )r s "r r r m s4 3P 3PuS$Y / 3Px} 3PPT 3Pr c h a ^ RI Ho R V3R llpR R lp\ R R l4 p\ R, pVf
V^ , pV^, p\ WV4 V! 4 ;_uu_ 4 p\ \ P P W4 V4 \ P P VR 4 pV! V4 w rV! V4 p V U u. uF qV , NK
p
p \ P ! W3\ P P V \ 4 4 RRR4 R# u up i + ' g i R# ; i)
a1 Parse the devkit archive of the ImageNet2012 classification dataset and save
the meta information in a binary file.
Args:
root (str or ``pathlib.Path``): Root directory containing the devkit archive
file (str, optional): Name of devkit archive. Defaults to
'ILSVRC2012_devkit_t12.tar.gz'
Nc
V ^8 d QhR\ R\ \ \ \ 3, \ \ \ \ R3, 3, 3, /# )r devkit_rootr .)r rV rW int)r s "r r *parse_devkit_archive..__annotate__x sA , ,C ,E$sCx.$sERUWZRZOG[B\2\,] ,r c < < \ P P V R R4 pS
P VRR7 R, p\ \ V! 4 ^, p\
V4 UUu. uF w rEV^ 8X g K W$, NK ppp\ \ V! 4 R, w rgpV U u. uF p \ V P R4 4 NK pp \ Wg4 UU
u/ uF w rJWJbK p pp
\ Wx4 U
U u/ uF w rWbK p p
p W3# u uppi u up i u up
pi u up p
i )datazmeta.matT)
squeeze_mesynsets:N Nz, ) r r! r8 loadmatrX zipr, rV r )rg metafilemeta
nums_childrenr0 num_childrenidcsr) r( r1 r/ idx_to_wnidr. sios & r parse_meta_mat,parse_devkit_archive..parse_meta_matx s 77<< VZ@{{8{5i@ S$Z(+
3<]3Ka3K /c|_`O` 3Ka #CJ / 3 W7>?wt5D)*w?25d2BC2BYSsy2B C8;E8KL8K*$4:8KL++ b?CLs $ D4 D##D
D4
Dc F V ^8 d QhR\ R\ \ , /# )r rg r )r rX rh )r s "r r ri s 6 6s 6tCy 6r c \ P P V R R4 p\ V4 ;_uu_ 4 pVP 4 pRRR4 X Uu. uF p\ V4 NK up# + ' g i L+; iu upi )rk z&ILSVRC2012_validation_ground_truth.txtN)r r! r8 open readlinesrh )rg rU txtfhval_idcsval_idxs & r parse_val_groundtruth_txt7parse_devkit_archive..parse_val_groundtruth_txt sX ww||K1YZ
$ZZ5 (H ,45HG H55 Z5s A)A<) A9 c : V ^8 d QhR\ \ , /# rA )r r )r s "r r ri s # ## #r c 3 " \ P ! 4 p V x \ P ! V 4 R # \ P ! T 4 i ; i5irD )tempfilemkdtempshutilrmtree)tmp_dirs r get_tmp_dir)parse_devkit_archive..get_tmp_dir s5 ""$ #M MM' "FMM' "s A5 AA
Ar ILSVRC2012_devkit_t12) scipy.ioior ARCHIVE_METArb r r r! r8 r[ saver9 )r rU rx r r archive_metar` r rg rv r. r r0 val_wnidsrw s && @r r: r: m s , ,6 # # )L |A
q/CD$ ' T0':ggll7,CD '5k'B$ ,[919:#%% :
O/ dI1NO
;
s $A D D$parse_val_archive.. s% T?SeBGGLL511?Ss -0)r r$ rb r r! r8 r sortedr setmkdirrp r movebasename)
r rU r) r r r` imagesr/ img_filer s
&&&& @r r= r= s &L |A
q/C } t$Q'D$ww||D)HBGGLL,h7
Trzz(?ST
TFE
h-. e, Hbggll8T277;K;KH;U VW -r )zILSVRC2012_img_train.tar 1d675b47d978889d74fa0da5fadfb00e)zILSVRC2012_img_val.tar 29b22e2961454d5413ddabcf34fc5622)z ILSVRC2012_devkit_t12.tar.gz fa75699e90414af021442c21a62c3abfrD )Nr
)NNr ) r r r collections.abcr
contextlibr pathlibr typingr r r r[ r r utilsr
r r r r9 r r$ rb r: r< r= r6 r r r s{
$ % ' ' C C
M I R
;8{ ;8| 3 33PlU4!X !Xr