NotificationsYou must be signed in to change notification settings
Fork3k
Star14k

fix mosaic and autoaug#8533

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to ourterms of service andprivacy statement. We’ll occasionally send you account related emails.

Already on GitHub?Sign in to your account

Jump to bottom

Open

GuoQuanhao wants to merge2 commits intoPaddlePaddle:release/2.6

base:release/2.6

Choose a base branch

fromGuoQuanhao:mosaic-autoaug-fix

Open

fix mosaic and autoaug#8533

GuoQuanhao wants to merge2 commits intoPaddlePaddle:release/2.6fromGuoQuanhao:mosaic-autoaug-fix

Conversation

Copy link

GuoQuanhao commentedAug 13, 2023

见#8517
本次修改并测试的算子包括operators.py中的Mosaic，RandomDistort，AutoAugment
autoaugment_utils.py中的policy_v3，其中带only的算子例如TranslateY_Only_BBoxes，ShearY_Only_BBoxes并未纳入本次训练进程的测试中

修改完成的数据增强，通过双3090在COCO数据集通过picodet训练10个epoch得到如下精度

其中数据读取yml文件如下

worker_num: 0eval_height: &eval_height 640eval_width: &eval_width 640eval_size: &eval_size [*eval_height, *eval_width]TrainReader:  sample_transforms:  - Decode: {}  - Mosaic:      prob: 1.0      input_dim: [640, 640]      degrees: [-10, 10]      scale: [0.5, 1.5]      shear: [-0.5, 0.5]      translate: [-0.1, 0.1]      enable_mixup: True  - AutoAugment: {autoaug_type: 'v3'}  - DebugVisibleImage: {output_dir: '../debug'}  batch_transforms:  - BatchRandomResize: {target_size: [576, 608, 640, 672, 704], random_size: True, random_interp: True, keep_ratio: False}  - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}  - Permute: {}  - PadGT: {}  batch_size: 32  shuffle: true  mosaic_epoch: 10  drop_last: trueEvalReader:  sample_transforms:  - Decode: {}  - Resize: {interp: 2, target_size: *eval_size, keep_ratio: False}  - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}  - Permute: {}  batch_transforms:  - PadBatch: {pad_to_stride: 32}  batch_size: 8  shuffle: falseTestReader:  inputs_def:    image_shape: [1, 3, *eval_height, *eval_width]  sample_transforms:  - Decode: {}  - Resize: {interp: 2, target_size: *eval_size, keep_ratio: False}  - NormalizeImage: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]}  - Permute: {}  batch_size: 1