YhbddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZmZddlmZGd d eZGd d eZdS) N)Iterable)Path)AnyCallablecastOptionalUnion)Image)iterable_to_strverify_str_arg) VisionDatasetc xeZdZ d dedeedeeddffd Zdedee e ffdZ defd Z xZ S) LSUNClassNroot transformtarget_transformreturnc6ddl}t|||||ddddd|_|jd5}|d|_dddn #1swxYwYd d d |Dz}tj |r)tj t|d |_dS|jd5}d |ddD|_dddn #1swxYwYtj|jt|ddS)Nrrrr TF) max_readersreadonlylock readaheadmeminitwriteentries_cache_c36K|]}|tjv|VdSN)string ascii_letters).0cs k/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/datasets/lsun.py z%LSUNClass.__init__..s/(V(VqAAUz&LSUNClass.__init__..s[[[SS[[[r))keysvalueswb)lmdbsuper__init__openenvbeginstatlengthjoinospathisfilepickleloadr/cursoriternextdump)selfrrrr2txn cache_file __class__s r'r4zLSUNClass.__init__s   EUVVV99Tq4eW\fk9ll X^^%^ ( ( 0C((**Y/DK 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0(V(VD(V(V(V!V!VV 7>>* % % ; DT$:$:;;DIIIe,, \[[CJJLL,A,AtTY,A,Z,Z[[[  \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ K 4 D#9#9 : : : : :s$% BBB!9E&&E*-E*indexcd\}}|j}|d5}||j|}dddn #1swxYwYt j}|||dtj | d}|j | |}|j | |}||fS)NNNFrrRGB) r6r7getr/ioBytesIOrseekr r5convertrr)rCrGimgtargetr6rDimgbufbufs r' __getitem__zLSUNClass.__getitem__"s Vh YYUY # # /sWWTYu-..F / / / / / / / / / / / / / / /jll &  joo%%e,, > %..%%C  ,**622FF{s!AAAc|jSr"r9rCs r'__len__zLSUNClass.__len__5 {r)rI) __name__ __module__ __qualname__strrrr4inttuplerrTrX __classcell__rFs@r'rrsfj;;;$,X$6;QYZbQc; ;;;;;;$sCx&r)rc eZdZdZ ddeeefdeeeefdee dee ddf fd Z deeeefdeefd Z d e de eeffd Zde fd ZdefdZxZS)LSUNa`LSUN `_ dataset. You will need to install the ``lmdb`` package to use this dataset: run ``pip install lmdb`` Args: root (str or ``pathlib.Path``): Root directory for the database files. classes (string or list): One of {'train', 'val', 'test'} or a list of categories to load. e,g. ['bedroom_train', 'church_outdoor_train']. 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 the target and transforms it. trainNrclassesrrrc t||||||_g|_|jD]L}|jt tj ||d|Mg|_ d}|jD].}|t|z }|j |/||_ dS)Nr_lmdb)rrr) r3r4_verify_classesredbsappendrr;r<r:indiceslenr9) rCrrerrr&countdbrFs r'r4z LSUN.__init__Is EUVVV++G44  b bA HOOI27<<kkk+J+JV_``` a a a a ( ' 'B SWW E L   & & & & r)c &gd}gd} tttd|dkrgnfd|DnF#t$r8t t s1d}t|ttd}D]}t||t|| d }d |dd |d } }d } | |d t|}t||| | | dt|}t| || YnwxYwS)N) bedroombridgechurch_outdoor classroomconference_room dining_roomkitchen living_room restauranttower)rdvaltestrer{c g|] }|dzz S)_r,)r%r&res r'r.z(LSUN._verify_classes..us"AAA1s7W,AAAr)zDExpected type str or Iterable for argument classes, but got type {}.zDExpected type str for elements in argument classes, but got type {}.) custom_msgr}z3Unknown value '{}' for {}. Valid values are {{{}}}.z LSUN class) valid_valuesr~postfix) rr]r ValueError isinstancerformattypelistsplitr:r ) rCre categories dset_optsmsgmsg_fmtstr_typer&c_shortcategorydset_opt msg_fmtstrs ` r'rhzLSUN._verify_classes`s    -,,  Q3((G 7Iy 9 9 9&  ")AAAAjAAA Q Q Qgx00 <\ DMM!:!:;;;7mmGdO Q Qq_-C-CDGG-L-LMMMM''#,,%(XXgcrcl%;%;WR[(R  '',PZ@[@[\\xjSQQQQ '')_Y=W=WXXxiCPPPPP Q Q Q&s>A EF FrGcd}d}|jD]}||krn|dz }|}|j|}||z }|j||}||\}}||fS)z Args: index (int): Index Returns: tuple: Tuple (image, target) where target is the index of the target category. rr )rkrir)rCrGrQsubindrnrPr}s r'rTzLSUN.__getitem__s<  Cs{{ aKFCC Xf    ,**622FEQF{r)c|jSr"rVrWs r'rXz LSUN.__len__rYr)c&djdi|jS)NzClasses: {classes}r,)r__dict__rWs r' extra_reprzLSUN.extra_reprs*#*;;T];;;r))rdNN)rZr[r\__doc__r r]rrrrr4rhr^r_rrTrXrr`ras@r'rcrc9s;  $*1(,/3 CIsDI~&H%  #8,   .)uS$s)^'<)c))))VsCx2r#collections.abcrpathlibrtypingrrrrr PILr utilsr r visionrrrcr,r)r'rs  $$$$$$7777777777777722222222!!!!!!''''' '''To<o<o<o<o<=o<o<o<o<o