From 9665d51a4988279f3d1aa98ec93d33c4af11303b Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Sat, 11 Jul 2020 08:58:14 +0000 Subject: [PATCH] beauty tutorials' code --- tutorials/train/image_classification/alexnet.py | 6 ++++-- .../train/image_classification/mobilenetv2.py | 6 ++++-- .../mobilenetv3_small_ssld.py | 6 ++++-- .../image_classification/resnet50_vd_ssld.py | 6 ++++-- .../train/image_classification/shufflenetv2.py | 6 ++++-- .../instance_segmentation/mask_rcnn_hrnet_fpn.py | 10 +++++----- .../instance_segmentation/mask_rcnn_r50_fpn.py | 14 +++++++------- .../object_detection/faster_rcnn_hrnet_fpn.py | 14 +++++++------- .../object_detection/faster_rcnn_r50_fpn.py | 10 +++++----- .../train/object_detection/yolov3_darknet53.py | 16 +++++++--------- .../train/object_detection/yolov3_mobilenetv1.py | 6 ++---- .../train/object_detection/yolov3_mobilenetv3.py | 16 +++++++--------- .../deeplabv3p_mobilenetv2.py | 9 ++++++--- .../train/semantic_segmentation/fast_scnn.py | 9 ++++++--- tutorials/train/semantic_segmentation/hrnet.py | 9 ++++++--- tutorials/train/semantic_segmentation/unet.py | 6 ++++-- 16 files changed, 82 insertions(+), 67 deletions(-) diff --git a/tutorials/train/image_classification/alexnet.py b/tutorials/train/image_classification/alexnet.py index 90819a9..b78f45b 100644 --- a/tutorials/train/image_classification/alexnet.py +++ b/tutorials/train/image_classification/alexnet.py @@ -7,12 +7,14 @@ pdx.utils.download_and_decompress(veg_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomCrop(crop_size=224), transforms.RandomHorizontalFlip(), + transforms.RandomCrop(crop_size=224), + transforms.RandomHorizontalFlip(), transforms.Normalize() ]) eval_transforms = transforms.Compose([ transforms.ResizeByShort(short_size=256), - transforms.CenterCrop(crop_size=224), transforms.Normalize() + transforms.CenterCrop(crop_size=224), + transforms.Normalize() ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/image_classification/mobilenetv2.py b/tutorials/train/image_classification/mobilenetv2.py index bc30dfd..5edbf58 100644 --- a/tutorials/train/image_classification/mobilenetv2.py +++ b/tutorials/train/image_classification/mobilenetv2.py @@ -8,12 +8,14 @@ pdx.utils.download_and_decompress(veg_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomCrop(crop_size=224), transforms.RandomHorizontalFlip(), + transforms.RandomCrop(crop_size=224), + transforms.RandomHorizontalFlip(), transforms.Normalize() ]) eval_transforms = transforms.Compose([ transforms.ResizeByShort(short_size=256), - transforms.CenterCrop(crop_size=224), transforms.Normalize() + transforms.CenterCrop(crop_size=224), + transforms.Normalize() ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/image_classification/mobilenetv3_small_ssld.py b/tutorials/train/image_classification/mobilenetv3_small_ssld.py index b8e5e43..6d52775 100644 --- a/tutorials/train/image_classification/mobilenetv3_small_ssld.py +++ b/tutorials/train/image_classification/mobilenetv3_small_ssld.py @@ -8,12 +8,14 @@ pdx.utils.download_and_decompress(veg_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomCrop(crop_size=224), transforms.RandomHorizontalFlip(), + transforms.RandomCrop(crop_size=224), + transforms.RandomHorizontalFlip(), transforms.Normalize() ]) eval_transforms = transforms.Compose([ transforms.ResizeByShort(short_size=256), - transforms.CenterCrop(crop_size=224), transforms.Normalize() + transforms.CenterCrop(crop_size=224), + transforms.Normalize() ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/image_classification/resnet50_vd_ssld.py b/tutorials/train/image_classification/resnet50_vd_ssld.py index 97a2e31..ca94f7f 100644 --- a/tutorials/train/image_classification/resnet50_vd_ssld.py +++ b/tutorials/train/image_classification/resnet50_vd_ssld.py @@ -8,12 +8,14 @@ pdx.utils.download_and_decompress(veg_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomCrop(crop_size=224), transforms.RandomHorizontalFlip(), + transforms.RandomCrop(crop_size=224), + transforms.RandomHorizontalFlip(), transforms.Normalize() ]) eval_transforms = transforms.Compose([ transforms.ResizeByShort(short_size=256), - transforms.CenterCrop(crop_size=224), transforms.Normalize() + transforms.CenterCrop(crop_size=224), + transforms.Normalize() ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/image_classification/shufflenetv2.py b/tutorials/train/image_classification/shufflenetv2.py index 244c434..29272df 100644 --- a/tutorials/train/image_classification/shufflenetv2.py +++ b/tutorials/train/image_classification/shufflenetv2.py @@ -8,12 +8,14 @@ pdx.utils.download_and_decompress(veg_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomCrop(crop_size=224), transforms.RandomHorizontalFlip(), + transforms.RandomCrop(crop_size=224), + transforms.RandomHorizontalFlip(), transforms.Normalize() ]) eval_transforms = transforms.Compose([ transforms.ResizeByShort(short_size=256), - transforms.CenterCrop(crop_size=224), transforms.Normalize() + transforms.CenterCrop(crop_size=224), + transforms.Normalize() ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/instance_segmentation/mask_rcnn_hrnet_fpn.py b/tutorials/train/instance_segmentation/mask_rcnn_hrnet_fpn.py index 920c5f2..7f8e1eb 100644 --- a/tutorials/train/instance_segmentation/mask_rcnn_hrnet_fpn.py +++ b/tutorials/train/instance_segmentation/mask_rcnn_hrnet_fpn.py @@ -11,15 +11,15 @@ pdx.utils.download_and_decompress(xiaoduxiong_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), transforms.Padding(coarsest_stride=32) + transforms.RandomHorizontalFlip(), + transforms.Normalize(), + transforms.ResizeByShort(short_size=800, max_size=1333), + transforms.Padding(coarsest_stride=32) ]) eval_transforms = transforms.Compose([ transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), + transforms.ResizeByShort(short_size=800, max_size=1333), transforms.Padding(coarsest_stride=32), ]) diff --git a/tutorials/train/instance_segmentation/mask_rcnn_r50_fpn.py b/tutorials/train/instance_segmentation/mask_rcnn_r50_fpn.py index bacbe61..1d9510f 100644 --- a/tutorials/train/instance_segmentation/mask_rcnn_r50_fpn.py +++ b/tutorials/train/instance_segmentation/mask_rcnn_r50_fpn.py @@ -11,16 +11,16 @@ pdx.utils.download_and_decompress(xiaoduxiong_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), transforms.Padding(coarsest_stride=32) + transforms.RandomHorizontalFlip(), + transforms.Normalize(), + transforms.ResizeByShort(short_size=800, max_size=1333), + transforms.Padding(coarsest_stride=32) ]) eval_transforms = transforms.Compose([ - transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), - transforms.Padding(coarsest_stride=32), + transforms.Normalize(), + transforms.ResizeByShort(short_size=800, max_size=1333), + transforms.Padding(coarsest_stride=32) ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/object_detection/faster_rcnn_hrnet_fpn.py b/tutorials/train/object_detection/faster_rcnn_hrnet_fpn.py index 83f9b69..3d1650f 100644 --- a/tutorials/train/object_detection/faster_rcnn_hrnet_fpn.py +++ b/tutorials/train/object_detection/faster_rcnn_hrnet_fpn.py @@ -11,16 +11,16 @@ pdx.utils.download_and_decompress(insect_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), transforms.Padding(coarsest_stride=32) + transforms.RandomHorizontalFlip(), + transforms.Normalize(), + transforms.ResizeByShort(short_size=800, max_size=1333), + transforms.Padding(coarsest_stride=32) ]) eval_transforms = transforms.Compose([ - transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), - transforms.Padding(coarsest_stride=32), + transforms.Normalize(), + transforms.ResizeByShort(short_size=800, max_size=1333), + transforms.Padding(coarsest_stride=32) ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/object_detection/faster_rcnn_r50_fpn.py b/tutorials/train/object_detection/faster_rcnn_r50_fpn.py index 6c7987a..ca41058 100644 --- a/tutorials/train/object_detection/faster_rcnn_r50_fpn.py +++ b/tutorials/train/object_detection/faster_rcnn_r50_fpn.py @@ -8,15 +8,15 @@ pdx.utils.download_and_decompress(insect_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), transforms.Padding(coarsest_stride=32) + transforms.RandomHorizontalFlip(), + transforms.Normalize(), + transforms.ResizeByShort(short_size=800, max_size=1333), + transforms.Padding(coarsest_stride=32) ]) eval_transforms = transforms.Compose([ transforms.Normalize(), - transforms.ResizeByShort( - short_size=800, max_size=1333), + transforms.ResizeByShort(short_size=800, max_size=1333), transforms.Padding(coarsest_stride=32), ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/object_detection/yolov3_darknet53.py b/tutorials/train/object_detection/yolov3_darknet53.py index facb3e5..a15e5cb 100644 --- a/tutorials/train/object_detection/yolov3_darknet53.py +++ b/tutorials/train/object_detection/yolov3_darknet53.py @@ -8,20 +8,18 @@ pdx.utils.download_and_decompress(insect_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.MixupImage(mixup_epoch=250), + transforms.MixupImage(mixup_epoch=250), transforms.RandomDistort(), - transforms.RandomExpand(), - transforms.RandomCrop(), - transforms.Resize( - target_size=608, interp='RANDOM'), + transforms.RandomExpand(), + transforms.RandomCrop(), + transforms.Resize(target_size=608, interp='RANDOM'), transforms.RandomHorizontalFlip(), - transforms.Normalize(), + transforms.Normalize() ]) eval_transforms = transforms.Compose([ - transforms.Resize( - target_size=608, interp='CUBIC'), - transforms.Normalize(), + transforms.Resize(target_size=608, interp='CUBIC'), + transforms.Normalize() ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/object_detection/yolov3_mobilenetv1.py b/tutorials/train/object_detection/yolov3_mobilenetv1.py index c745726..9b62185 100644 --- a/tutorials/train/object_detection/yolov3_mobilenetv1.py +++ b/tutorials/train/object_detection/yolov3_mobilenetv1.py @@ -12,15 +12,13 @@ train_transforms = transforms.Compose([ transforms.RandomDistort(), transforms.RandomExpand(), transforms.RandomCrop(), - transforms.Resize( - target_size=608, interp='RANDOM'), + transforms.Resize(target_size=608, interp='RANDOM'), transforms.RandomHorizontalFlip(), transforms.Normalize(), ]) eval_transforms = transforms.Compose([ - transforms.Resize( - target_size=608, interp='CUBIC'), + transforms.Resize(target_size=608, interp='CUBIC'), transforms.Normalize(), ]) diff --git a/tutorials/train/object_detection/yolov3_mobilenetv3.py b/tutorials/train/object_detection/yolov3_mobilenetv3.py index 3fab525..4eb0609 100644 --- a/tutorials/train/object_detection/yolov3_mobilenetv3.py +++ b/tutorials/train/object_detection/yolov3_mobilenetv3.py @@ -8,20 +8,18 @@ pdx.utils.download_and_decompress(insect_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.MixupImage(mixup_epoch=250), + transforms.MixupImage(mixup_epoch=250), transforms.RandomDistort(), - transforms.RandomExpand(), - transforms.RandomCrop(), - transforms.Resize( - target_size=608, interp='RANDOM'), + transforms.RandomExpand(), + transforms.RandomCrop(), + transforms.Resize(target_size=608, interp='RANDOM'), transforms.RandomHorizontalFlip(), - transforms.Normalize(), + transforms.Normalize() ]) eval_transforms = transforms.Compose([ - transforms.Resize( - target_size=608, interp='CUBIC'), - transforms.Normalize(), + transforms.Resize(target_size=608, interp='CUBIC'), + transforms.Normalize() ]) # 定义训练和验证所用的数据集 diff --git a/tutorials/train/semantic_segmentation/deeplabv3p_mobilenetv2.py b/tutorials/train/semantic_segmentation/deeplabv3p_mobilenetv2.py index 9e3a7d5..d7a2f63 100644 --- a/tutorials/train/semantic_segmentation/deeplabv3p_mobilenetv2.py +++ b/tutorials/train/semantic_segmentation/deeplabv3p_mobilenetv2.py @@ -11,12 +11,15 @@ pdx.utils.download_and_decompress(optic_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.ResizeRangeScaling(), - transforms.RandomPaddingCrop(crop_size=512), transforms.Normalize() + transforms.RandomHorizontalFlip(), + transforms.ResizeRangeScaling(), + transforms.RandomPaddingCrop(crop_size=512), + transforms.Normalize() ]) eval_transforms = transforms.Compose([ - transforms.ResizeByLong(long_size=512), transforms.Padding(target_size=512), + transforms.ResizeByLong(long_size=512), + transforms.Padding(target_size=512), transforms.Normalize() ]) diff --git a/tutorials/train/semantic_segmentation/fast_scnn.py b/tutorials/train/semantic_segmentation/fast_scnn.py index a2fdfa4..af041ca 100644 --- a/tutorials/train/semantic_segmentation/fast_scnn.py +++ b/tutorials/train/semantic_segmentation/fast_scnn.py @@ -12,12 +12,15 @@ pdx.utils.download_and_decompress(optic_dataset, path='./') # 定义训练和验证时的transforms # API说明: https://paddlex.readthedocs.io/zh_CN/latest/apis/transforms/seg_transforms.html#composedsegtransforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.ResizeRangeScaling(), - transforms.RandomPaddingCrop(crop_size=512), transforms.Normalize() + transforms.RandomHorizontalFlip(), + transforms.ResizeRangeScaling(), + transforms.RandomPaddingCrop(crop_size=512), + transforms.Normalize() ]) eval_transforms = transforms.Compose([ - transforms.ResizeByLong(long_size=512), transforms.Padding(target_size=512), + transforms.ResizeByLong(long_size=512), + transforms.Padding(target_size=512), transforms.Normalize() ]) diff --git a/tutorials/train/semantic_segmentation/hrnet.py b/tutorials/train/semantic_segmentation/hrnet.py index 682ce82..330a107 100644 --- a/tutorials/train/semantic_segmentation/hrnet.py +++ b/tutorials/train/semantic_segmentation/hrnet.py @@ -11,12 +11,15 @@ pdx.utils.download_and_decompress(optic_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.ResizeRangeScaling(), - transforms.RandomPaddingCrop(crop_size=512), transforms.Normalize() + transforms.RandomHorizontalFlip(), + transforms.ResizeRangeScaling(), + transforms.RandomPaddingCrop(crop_size=512), + transforms.Normalize() ]) eval_transforms = transforms.Compose([ - transforms.ResizeByLong(long_size=512), transforms.Padding(target_size=512), + transforms.ResizeByLong(long_size=512), + transforms.Padding(target_size=512), transforms.Normalize() ]) diff --git a/tutorials/train/semantic_segmentation/unet.py b/tutorials/train/semantic_segmentation/unet.py index 327a6ce..46e93b8 100644 --- a/tutorials/train/semantic_segmentation/unet.py +++ b/tutorials/train/semantic_segmentation/unet.py @@ -11,8 +11,10 @@ pdx.utils.download_and_decompress(optic_dataset, path='./') # 定义训练和验证时的transforms train_transforms = transforms.Compose([ - transforms.RandomHorizontalFlip(), transforms.ResizeRangeScaling(), - transforms.RandomPaddingCrop(crop_size=512), transforms.Normalize() + transforms.RandomHorizontalFlip(), + transforms.ResizeRangeScaling(), + transforms.RandomPaddingCrop(crop_size=512), + transforms.Normalize() ]) eval_transforms = transforms.Compose([ -- GitLab