gSrSSKJs Jr SSKrSSKrSSKrSSKJr SSK J r \R"\ 5r Sr"SS\R5r"S S \R"5r"S S \RRR5rg) zfQuick n Simple Image Folder, Tarfile based DataSet Hacked together by / Copyright 2020 Ross Wightman N)Image) create_parser2cF\rSrSrS SjrSrSrS SjrS SjrSr g) ImageDatasetNcUb[U[5(a[U=(d SXS9nX lX@lXPlSUlg)N)root class_mapr) isinstancestrrparser load_bytes transform_consecutive_errors)selfr rr rrs 6C:\Users\Terry\Desktop\FGVC-HERBS\timm\data\dataset.py__init__ImageDataset.__init__s< >Z44"6> %..%C >\\"EJJ7F{  OO4UG74;;CWCWX]C^B__bcfghcibjk l  $ $ ) $'',6''c$++6F(FGG  s%AB'' E1BE E E  Ec,[UR5$N)r'rrs r__len__ImageDataset.__len__7s4;;rc:URRXU5$r1)rr$rr+basenameabsolutes rr$ImageDataset.filename:s{{##EX>>rc8URRX5$r1r filenamesrr7r8s rr<ImageDataset.filenames={{$$X88r)rrrr)Nr FNFF) __name__ __module__ __qualname____firstlineno__rr&r3r$r<__static_attributes__rrrrs)  %$ ?9rrcN\rSrSrS SjrSrSrS SjrS SjrSr g) IterableImageDatasetANc Uce[U[5(a[X!X4XXS9UlOX lXlSUlg)N)r split is_training batch_sizerepeatsr)rrrrrr) rr rrKrLrMr rrNrs rrIterableImageDataset.__init__CsF!!! fc " "'T^qDK!K"#$ rc## URHNupURbURU5nUc#[R"S[RS9nX4v MP g7f)Nrr)rrr(r)r*)rr,r-s r__iter__IterableImageDataset.__iter__XsN;;KC~~)nnS)~b ;+  'sA A"cd[URS5(a[UR5$g)Nr3r)hasattrrr'r2s rr3IterableImageDataset.__len__`s% 4;; * *t{{# #rcS5e)Nz8Filename lookup by index not supported, use filenames().rFr6s rr$IterableImageDataset.filenamefs PPPurc8URRX5$r1r;r=s rr<IterableImageDataset.filenamesir?r)rrr)NtrainFNr FrNr@) rArBrCrDrrQr3r$r<rErFrrrHrHAs6 %* Q9rrHcn\rSrSrSrS SjrSr\S5r\RS5rSr Sr S r S r g ) AugMixDatasetmzCDataset wrapper to perform AugMix or other clean/augmentation mixescSUlSUlXlURRb%UR URR5 X lgr1) augmentation normalizedatasetr_set_transforms num_splits)rrarcs rrAugMixDataset.__init__psC  << ! ! -  !7!7 8$rc[U[[45(a[U5S:XdS5eUSURlUSUlUSUlg)Nz&Expecting a tuple/list of 3 transformsrr)rlisttupler'rarr_r`rxs rrbAugMixDataset._set_transformsxsP!dE]++A! e=ee;!"1 aD1rc.URR$r1)rarr2s rrAugMixDataset.transform~s||%%%rc&URU5 gr1)rbrjs rrrns QrcBURcU$URU5$r1)r`rjs r _normalizeAugMixDataset._normalizesNN*qAq0AArcURUup#URU5/n[URS- 5H2nUR URUR U555 M4 [ U5U4$)Nr)rarqrangercappendr_ri)rirkyx_list_s rr&AugMixDataset.__getitem__si||A//!$%t*+A MM$//$*;*;A*>? @,V}arc,[UR5$r1)r'rar2s rr3AugMixDataset.__len__s4<<  r)r_rar`rcN)rg)rArBrCrD__doc__rrbpropertyrsetterrqr&r3rErFrrr\r\msMM% &&  B !rr\)r}torch.utils.datautilsdataosr(loggingPILrparsersr getLoggerrAr"r%DatasetrIterableDatasetrHr\rFrrrsx  "   H % *94<<*9Z)94//)9X%!EKK$$,,%!r