U
|
è³æ` ã @ s. d dl mZ ddlT ddlmZ dddZdS ) é Né )Ú*)Ú AutoAugmentTc C sÜ g }|r| j j}| j j}| j j}| jj}| j j}| j j}| j j} | j j }
|
| j j
|
} | j j j } | j j j } | j j j}| j j j}| j j j}| j j j}| j jj }| j jj }| j jj}| j jj }| j jj }|rÒ| t|¡ | tj|dd¡ |rü| tj|d¡ | r | tj|
|
| dt |¡g¡ | rH| tj t ||||¡g| d¡ |rZ| t ¡ |rr| t ||d¡ |r¦| t!|d¡ n | j j"}| tj|dd¡ | t# ¡ | tj$ddd gd
|
d d gd ¡ t %|¡S )Né )Ú interpolation)Úp)Ú padding_mode)Z probabilityÚmean)Z prob_happeng33333ë^@gR¸
ë]@gR¸
ëáY@gÃõ(\2M@gÂõ(\L@g °L@)r Ústd)&ÚINPUTÚ
|
SIZE_TRAINÚ DO_AUGMIXÚ
|
DO_AUTOAUGÚSOLVERÚMAX_ITERÚDO_FLIPÚ FLIP_PROBÚDO_PADÚPADDINGÚ PADDING_MODEÚCJÚENABLEDÚPROBÚ
|
BRIGHTNESSÚCONTRASTÚ
|
SATURATIONÚHUEÚREAÚMEANÚRPTÚappendr ÚTÚResizeZRandomHorizontalFlipÚextendZPadZ
|
RandomCropZ RandomApplyZ ColorJitterZAugMixZ RandomErasingZ RandomPatchÚ SIZE_TESTÚToTensorZ NormalizeÚCompose)ÚcfgÚis_trainÚresZ
|
size_trainZ do_augmixZ
|
do_autoaugZ
|
total_iterZdo_flipZ flip_probZdo_padÚpaddingr Zdo_cjZcj_probZ cj_brightnessZ cj_contrastZ cj_saturationZcj_hueZdo_reaZrea_probZrea_meanZdo_rptZrpt_probZ size_test© r+ ú\/data/disk1/workspace/06_reid/01_fast_reid/02_fast_reid_inference/./data/transforms/build.pyÚbuild_transforms sX
|
|
|
|
|
|
|
|
|
|
|
ÿ" r- )T)Útorchvision.transformsÚ
|
transformsr! Ú autoaugmentr r- r+ r+ r+ r, Ú<module> s
|
