RЦiS?FSrSSKrSSKJrJrJr SSKrSSKJr SSK J r J r J r SSK JrJrJr SSKJrJrJrJrJrJrJrJrJr SSKJrJrJr SS KJ r S S \ \ S S 4S\\!\\!\!44S\"S\\#S4S\\#S4S\$S\$4 Sjjr%S SSSSSSSS SSSS\ \ SSSSS S S S SSS 4S\\!\\!\!44S\\\#\#4S\\\#\#4S \\"S!\#S"\#S#\\#\\#S44S$\\#S%\$S&\#S'\#S(\\"S\"S\\#S4S\\#S4S)\#S*\"S+\!S,\!S\$S\$S-\$S.\$S/\\!\\!\!44S0\!S1\$44S2jjr&S SSSS \ \ S S S SSS 4 S\\!\\!\!44S3\\#S4\\"S5\\!S\"S\\#S4S\\#S4S\$S\$S.\$S/\\!\\!\!44S0\!S1\$4S6jjr'S S S SSSSSSSSSSS \ \ SSSSSSSS S S S S SSS 4S7\\!\\!\!4\\!\!\!44S8\$S9\$S \\"S\\\#\#4S\\\#\#4S!\#S"\#S#\\#\\#S44S$\\#S&\#S'\#S(\\"S\"S\\#S4S\\#S4S)\#S*\"S+\!S,\!S3\\#S4\\"S5\\!S:\$S\$S\$S-\$S.\$S/\\!\\!\!44S0\!S1\$4>S;jjr(g)"U 5[<R>"U5S!9/- n US:aU [AUUUUS"S#9/- n U(a U [CUS$9/- n U(aA[ RD"U5[ RD"U5[ RD"U 54$[ RD"UU-U -5$s snf)%a#ImageNet-oriented image transforms for training. Args: img_size: Target image size. train_crop_mode: Training random crop mode ('rrc', 'rkrc', 'rkrr'). scale: Random resize scale range (crop area, < 1.0 => zoom in). ratio: Random aspect ratio range (crop ratio for RRC, ratio adjustment factor for RKR). hflip: Horizontal flip probability. vflip: Vertical flip probability. color_jitter: Random color jitter component factors (brightness, contrast, saturation, hue). Scalar is applied as (scalar,) * 3 (no hue). color_jitter_prob: Apply color jitter with this probability if not None (for SimlCLR-like aug). force_color_jitter: Force color jitter where it is normally disabled (ie with RandAugment on). grayscale_prob: Probability of converting image to grayscale (for SimCLR-like aug). gaussian_blur_prob: Probability of applying gaussian blur (for SimCLR-like aug). auto_augment: Auto augment configuration string (see auto_augment.py). interpolation: Image interpolation mode. mean: Image normalization mean. std: Image normalization standard deviation. re_prob: Random erasing probability. re_mode: Random erasing fill mode. re_count: Number of random erasing regions. re_num_splits: Control split of random erasing across batch size. use_prefetcher: Prefetcher enabled. Do not convert image to tensor or normalize. normalize: Normalize tensor output w/ provided mean/std (if prefetcher not used). separate: Output transforms in 3-stage tuple. naflex: Enable NaFlex mode, sequence constrained patch output patch_size: Patch size for NaFlex mode. max_seq_len: Max sequence length for NaFlex mode. Returns: If separate==True, the transforms are returned as a tuple of 3 separate transforms for use in a mixing dataset that passes * all data through the first (primary) transform, called the 'clean' data * a portion of the data through the secondary transform * normalizes and converts the branches above with the third, final transform rrc>rJrkrcrkrr)g{Gz??)g?gUUUUUU?)rFrGr5r6r)rKrL)g?rM)g?grq?r0T)rrandom_scale_probrandom_scale_rangerandom_scale_arearandom_aspect_probrandom_aspect_rangerKreflect) padding_mode)r5r6rr1)pF3ag?)translate_constimg_meanr#rrandaugmixg333333? translate_pct)r]) kernel_sizer%cpu)mode max_count num_splitsdevicerF)#tuplerrrrrrRandomHorizontalFlipRandomVerticalFlip isinstancestrlistmindictintroundr startswithr r r lenfloat RandomApply ColorJitterRandomGrayscale GaussianBlurrrr(r)r*rrr+)!rr5r6r7r8r9r:r;r<r=r>r?rrrr@rArBrCr r!rDrErFrGrH primary_tfl secondary_tfldisable_color_jitter img_size_minx aa_params final_tfls! r-transforms_imagenet_trainrAskB&.O 5 55 5K e*{+e1123!#'     . .%.;/E%.;/E "/&)',&*'*(-#f, yAyA  K%.;/E%5#56E 1"/  K rz 77%@AA  rz 55>?? M ,,,,,%7$N$,:NO h . .x=L#L t 34dCdCU37^4dCD  ]h6):=)IIo &  " "6 * * 4\9MN NM  $ $X . .),Io & 7 iPQ QM 4\9MN NM(< lT5M 2 2|$. ..",/1A5L  ( &&".. =(( M j44lCD DM*44~FGG   " "++;$%   I&()) &())  O  \\$'LL%    R<  &,  Ih*566 !!+. 0B0B=0QS]SeSefoSppp!!+ "= "IJJYDs "P.crop_pct crop_modecrop_border_pixelsc pU=(d [n[U[[45(aD[ U5S:Xde[UV s/sHn [ R "X- 5PM sn 5nO[ R "X- 5nX4n/nU(aU[U5/- nU (aU[U U US9/- nOUS:Xa8U[R"U[U5S9[R"U5/- nOUS:Xa6UVs/sHn[SU-5PM nnU[XSS9[UUS 9/- nO[US US :Xa&U[R"US [U5S9/- nOU[U5/- nU[R"U5/- nU(aU[!5/- nO`U(dU[!5/- nOJU[#5[R$"[&R("U5[&R("U5S 9/- nU (a U[+U S 9/- n[R,"U5$s sn fs snf)aImageNet-oriented image transform for evaluation and inference. Args: img_size: Target image size. crop_pct: Crop percentage. Defaults to 0.875 when None. crop_mode: Crop mode. One of ['squash', 'border', 'center']. Defaults to 'center' when None. crop_border_pixels: Trim a border of specified # pixels around edge of original image. interpolation: Image interpolation mode. mean: Image normalization mean. std: Image normalization standard deviation. use_prefetcher: Prefetcher enabled. Do not convert image to tensor or normalize. normalize: Normalize tensor output w/ provided mean/std (if prefetcher not used). naflex: Enable NaFlex mode, sequence constrained patch output patch_size: Patch size for NaFlex mode. max_seq_len: Max sequence length for NaFlex mode. patchify: Patchify the output instead of relying on prefetcher Returns: Composed transform pipeline )rFrGrsquashr$borderrWrM)rlongest)fillrr3r%rf)r rjrgrlrrmathfloorrrrr&r r'rprrrrr(r)r*rr+)rrrrrrrr r!rErFrGrHr| scale_sizer,vrs r-transforms_imagenet_evalrs7F++H(UDM**8}!!!hGhDJJq|4hGH ZZ 34  - C -.//   !#'      !!* zoom in). ratio: Random aspect ratio range (crop ratio for RRC, ratio adjustment factor for RKR). hflip: Horizontal flip probability. vflip: Vertical flip probability. color_jitter: Random color jitter component factors (brightness, contrast, saturation, hue). Scalar is applied as (scalar,) * 3 (no hue). color_jitter_prob: Apply color jitter with this probability if not None (for SimlCLR-like aug). grayscale_prob: Probability of converting image to grayscale (for SimCLR-like aug). gaussian_blur_prob: Probability of applying gaussian blur (for SimCLR-like aug). auto_augment: Auto augment configuration string (see auto_augment.py). interpolation: Image interpolation mode. mean: Image normalization mean. std: Image normalization standard deviation. re_prob: Random erasing probability. re_mode: Random erasing fill mode. re_count: Number of random erasing regions. re_num_splits: Control split of random erasing across batch size. crop_pct: Inference crop percentage (output size / resize size). crop_mode: Inference crop mode. One of ['squash', 'border', 'center']. Defaults to 'center' when None. crop_border_pixels: Inference crop border of specified # pixels around edge of original image. tf_preprocessing: Use TF 1.0 inference preprocessing for testing model ports use_prefetcher: Pre-fetcher enabled. Do not convert image to tensor or normalize. normalize: Normalization tensor output w/ provided mean/std (if prefetcher not used). separate: Output transforms in 3-stage tuple. Returns: Composed transforms or tuple thereof Nz6Separate transforms not supported for TF preprocessingr)TfPreprocessTransform)rsizerz-Cannot perform split augmentation with no_aug)rrrr r!r7r5r6r8r9r:r;r=r>r?rrrr@rArBrCr r!rDrErFrGrHz>Separate transforms not supported for validation preprocessing) rrrrrrr r!rErFrGrH)rjrgrltimm.data.tf_preprocessingrr.rr)"rrrr7r5r6r8r9r:r;r=r>r?rrrr@rArBrCrrrrr r!rDrErFrGrHrr transforms" r-create_transformrysH*udm,,bc?NUUU|D)#' | q 6 P!P P<.+-# Il ]1 /     *#4 .$6*,   !""#$,%& .'($)*"+,-.&/0(12"3IZ # a!a a<0+!##5-#%'!I r/))__doc__rtypingrrrr) torchvisionrtimm.data.constantsrrr timm.data.auto_augmentr r r timm.data.transformsr rrrrrrrrtimm.data.naflex_transformsrrrtimm.data.random_erasingrrorkrsboolr.rrrr/r-rs  )) "]]llccc__214'"7!5$ *#U38_,-*#*#E3J*#5#:  *#  *#  *#\14/3/3)-8;-1#( "$&&*%"7!5$245LKU38_,-LKeUl+,LKeUl+,LK"# LK  LK  LKE5#445LK$E?LK!LKLK"LKsmLKLKE3JLK5#: LK !LK"#LK$%LK&'LK()LK*+LK,-LK./LK0#uS#X./1LK23LK45LK`14$(#',0'"7!5$24f#U38_,-f#5/f#C=f#%SM f#  f# E3J f#5#: f#f#f#f##uS#X./f#f#f#TIL!)-/3/38;-1 "$&&*'"7!5$(#',0!&$24?J#uS#Xc3m0DDEJJJ"# J eUl+, J eUl+, JJJE5#445J$E?JJ"JsmJJE3JJ 5#: !J"#J$%J&'J()J*5/+J,C=-J.%SM/J01J23J45J67J89J:#uS#X./;J<=J>?Jr/