Ph" R d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m
Z
m Z d dl Z ddl
mZ ddlmZmZmZ ddd d
Zd Z G d d
e Zddede
e de eeef e e f fdZdedededdfdZddede
e ddfdZddede
e deddfdZ ddede
e de
e e deddf
dZy) N)contextmanager)AnyDictIteratorListOptionalTuple ) ImageFolder)check_integrityextract_archiveverify_str_arg)zILSVRC2012_img_train.tar 1d675b47d978889d74fa0da5fadfb00e)zILSVRC2012_img_val.tar 29b22e2961454d5413ddabcf34fc5622)z ILSVRC2012_devkit_t12.tar.gz fa75699e90414af021442c21a62c3abf)trainvaldevkitzmeta.binc ` e Zd ZdZd dedededdf fd
Zd dZedefd Z defd
Z
xZ S )
ImageNetaN `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 (string): 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 an PIL image
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.
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
rootsplitkwargsreturnNc t j j | x}| _ t |dd | _ | j
t | j d }t | $ | j fi | || _ | j | _ | j | _ | j D cg c] }|| c}| _ t | j D ci c] \ }}|D ] }|| c}}}| _
y c c}w c c}}}w )Nr )r 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
hC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/datasets/imagenet.pyr" zImageNet.__init__0 s 77--d33ty#E74DE
(3A6
**5f5 \\
,,:>**E*$-*E 7@ 7N _7N)#tZ^SVS#XZ^S7N _ F _s 3 C? Dc t t j j | j t
st
| j t j j | j sK| j dk( rt | j y | j dk( rt | j y y y )Nr r ) r r r joinr META_FILEparse_devkit_archiveisdirr# r parse_train_archiveparse_val_archiver) s r0 r zImageNet.parse_archives? sy rww||DIIy AB +ww}}T . ./zzW$#DII.u$!$)), % 0 c j t j j | j | j S N)r r r2 r r r8 s r0 r# zImageNet.split_folderI s ww||DIItzz22r9 c : dj di | j S )NzSplit: {split} )format__dict__r8 s r0
extra_reprzImageNet.extra_reprM s & &&7 77r9 )r )r N) __name__
__module__ __qualname____doc__strr r" r propertyr# r@
__classcell__)r/ s @r0 r r sZ 4
`S
`
`#
`$
` - 3c 3 38C 8r9 r r filer c |t }t j j | | }t | rt j | S d}t |j || )NzThe meta file {} is not present in the root directory or is corrupted. This file is automatically created by the ImageNet dataset.) r3 r r r2 r torchload RuntimeErrorr> )r rH msgs r0 r r Q sZ |
77<<d #Dt zz$
J 3::dD122r9 md5c t t j j | | | s d}t |j || y )Nz{The archive {} is not present in the root directory or is corrupted. You need to download it externally and place it in {}.)r r r r2 rL r> )r rH rN rM s r0 _verify_archiverP ` sC 277<<d3S 9
E 3::dD122 :r9 c
ddl m dt dt t t
t f t t t t df f f ffd }dt dt t
fd}t dt t fd }t d }||d }|d
}t | || | 5 }t t j j | | | t j j |d } || \ } }
|| } | D cg c] } | | }
} t j |
|
ft j j | t" ddd yc c} w # 1 sw Y yxY w) a5 Parse the devkit archive of the ImageNet2012 classification dataset and save
the meta information in a binary file.
Args:
root (str): Root directory containing the devkit archive
file (str, optional): Name of devkit archive. Defaults to
'ILSVRC2012_devkit_t12.tar.gz'
r N devkit_rootr .c t j j | dd }
j |d d }t t | d }t
| D cg c] \ }}|dk( s || }}}t t | d d \ }}}|D cg c] } t | j d }} t || D
ci c] \ }}
||
} }}
t || D
ci c] \ }
} |
|
} }
} | | fS c c}}w c c} w c c}
}w c c} }
w )
Ndatazmeta.matT)
squeeze_mesynsets r z, ) r r r2 loadmatlistzipr( tupler )rR metafilemeta
nums_childrenr, num_childrenidcsr% r$ r- r+ idx_to_wnidr* sios r0 parse_meta_matz,parse_devkit_archive..parse_meta_matt s 77<< VZ@{{8{5i@ S$Z(+
3<]3Ka3K /c<|_`O`S 3Ka #CJ / 3 eW7>?wt5D)*w?25dE2BC2BYS$sDy2B C8;E78KL8K*$4:8KLO++ b?CLs
C4)C4!C:
C?
Dc t j j | dd }t | 5 }|j }d d d D cg c]
}t | c}S # 1 sw Y "xY wc c}w )NrT z&ILSVRC2012_validation_ground_truth.txt)r r r2 open readlinesint)rR rH txtfhval_idcsval_idxs r0 parse_val_groundtruth_txtz7parse_devkit_archive..parse_val_groundtruth_txt sV ww||K1YZ
$Z5 (H ,45HG H55 Z5s A
A+ A(c 3 K t j } | t j | y # t j | w xY wwr; )tempfilemkdtempshutilrmtree)tmp_dirs r0 get_tmp_dirz)parse_devkit_archive..get_tmp_dir s6 ""$ #M MM' "FMM' "s A 2 A A A r r
ILSVRC2012_devkit_t12)scipy.ioiorE r r rh r r r ARCHIVE_METArP r
r r r2 rJ saver3 )r rH rd rl rs archive_metarN rr rR rb r* rj r, val_wnidsrc s @r0 r4 r4 i sD ,C ,E$sCx.$sERUWZRZOG[B\2\,] ,6s 6tCy 6 ## # # )L |A
q/CD$$ ' T40':ggll7,CD '5k'B$ _,[919:#[% :
OY/ dI1NO
;
s $A"E Ez$parse_val_archive.. s# T?SeBGGLL51?Ss *-)rw r rP r r r2 r
sortedr~ setmkdirr[ rp movebasename)
r rH r% r{ ry rN imagesr+ img_filer s
@r0 r7 r7 s &L |A
q/C } t$Q'D$$ww||D&)HBGGLLt,h7
Trzz(?ST
TFE
h-. eV,h Hbggll8T277;K;KH;U VW -r9 r; )Nr )NNr ) r rp rn
contextlibr typingr r r r r r rJ r{ r utilsr r
r rw r3 r rE r rP r4 r6 r7 r= r9 r0 r s7
% = = C C N IR
98{ 98x 3 3HSM 3U4S>SWX[S\C\=] 3 3# 3S 3s 3t 33Ps 3P(3- 3P4 3PlUc U# Us UY] U6 ]b!X
!X c]!X2:492E!XVY!X !Xr9