DibddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZmZddlmZGd d eZGd d eZy) N)Iterable)Path)AnyCallablecastOptionalUnion)Image)iterable_to_strverify_str_arg) VisionDatasetc feZdZ d dedeedeeddffd Zdedee e ffdZ defd Z xZ S) LSUNClassNroot transformtarget_transformreturncddl}t| ||||j|ddddd|_|jj d5}|j d|_dddd d jd |Dz}tjj|r%tjt|d |_y|jj d5}|jj!dd Dcgc]}|c}|_dddtj"|jt|dy#1swYxYwcc}w#1swYExYw)Nrrrr TF) max_readersreadonlylock readaheadmeminitwriteentries_cache_c3FK|]}|tjvs|ywN)string ascii_letters).0cs [/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/lsun.py z%LSUNClass.__init__..s(VqAAUAU>* % DT$:;DIe, \,/JJL,A,AtTY,A,Z[SS[  \ KK 4 D#9 : 0 0\ \ \s*E,$E% E E%E E%%E.indexcd\}}|j}|jd5}|j|j|}dddt j }|j |jdtj|jd}|j|j|}|j|j|}||fS#1swYxYw)NNNFrrRGB) r1r2getr*ioBytesIOrseekr r0convertrr)r>rCimgtargetr1r?imgbufbufs r' __getitem__zLSUNClass.__getitem__"s Vhh YYUY # /sWWTYYu-.F /jjl &  jjo%%e, >> %..%C  ,**62FF{ / /s C##C,c|jSr"r4r>s r'__len__zLSUNClass.__len__5 {{rE) __name__ __module__ __qualname__strrrr/inttuplerrPrT __classcell__rBs@r'rrs]fj;;$,X$6;QYZbQc; ;$sCx&rVrc eZdZdZ ddeeefdeeeefdee dee ddf fd Z deeeefdeefd Z d e de eeffd Zde fd Zdefd ZxZS)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. Nrclassesrrrc t|||||j||_g|_|jD]I}|jj t tjj||d|Kg|_ d}|jD]+}|t|z }|jj |-||_ y)Nr_lmdb)rrr) r.r/_verify_classesradbsappendrr6r7r5indiceslenr4) r>rrarrr&countdbrBs r'r/z LSUN.__init__Is EUV++G4  bA HHOOI277<<5k+JV_` a b (( 'B SW E LL   & ' rVc gd}gd} tt|}t|d||dk(r|g}|S|Dcgc] }|dz|z }} |Scc}w#t$rt |t s%d}t|j t|t|}d}|D]}t||j t||jd}dj|dd |d } }d } | j |d t|}t||| | j | d t|}t| || Y|SwxYw)N) bedroombridgechurch_outdoor classroomconference_room dining_roomkitchen living_room restauranttower)trainvaltestrarx_zDExpected type str or Iterable for argument classes, but got type {}.zDExpected type str for elements in argument classes, but got type {}.) custom_msgz3Unknown value '{}' for {}. Valid values are {{{}}}.z LSUN class) valid_valuesrzpostfix) rrZr ValueError isinstancerformattypelistsplitr5r ) r>ra categories dset_optsr&msgmsg_fmtstr_typec_shortcategorydset_opt msg_fmtstrs r'rdzLSUN._verify_classes`s`  -  Q3(G 7Iy 9& "),)7AA1s7W,AA()B Qgx0\ DM!:;;7mGdO Qq_-C-CDG-LM''#,%(XXgcrl%;WR[(R  '',PZ@[\xjSQ '')_Y=WXxiCP Q' Qs(%AAA A AC7E E rCcd}d}|jD]}||krn |dz }|}|j|}||z }|j|j|}||\}}||fS)z Args: index (int): Index Returns: tuple: Tuple (image, target) where target is the index of the target category. rr )rgrer)r>rCrMsubindrjrLrys r'rPzLSUN.__getitem__s<< Cs{ aKFC   XXf    ,**62FEQF{rVc|jSr"rRrSs r'rTz LSUN.__len__rUrVc:djdi|jS)NzClasses: {classes})r__dict__rSs r' extra_reprzLSUN.extra_reprs*#**;T]];;rV)rvNN)rWrXrY__doc__r rZrrrrr/rdr[r\rrPrTrr]r^s@r'r`r`9s $*1(,/3 CIsDI~&H%  #8,   .)uS$s)^'<)c)VsCx2rsA  $772!' 'To<=o