From 662279771116cb693788d01974d9e4b0e11726c2 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Mon, 15 Nov 2021 07:26:58 +0000 Subject: [PATCH] fix some configs --- ppcls/configs/ImageNet/CSPNet/CSPDarkNet53.yaml | 2 +- ppcls/configs/ImageNet/DarkNet/DarkNet53.yaml | 12 ++++++------ .../ImageNet/EfficientNet/EfficientNetB1.yaml | 6 +++--- .../ImageNet/EfficientNet/EfficientNetB2.yaml | 6 +++--- .../ImageNet/EfficientNet/EfficientNetB3.yaml | 6 +++--- .../ImageNet/EfficientNet/EfficientNetB4.yaml | 6 +++--- .../ImageNet/EfficientNet/EfficientNetB5.yaml | 6 +++--- .../ImageNet/EfficientNet/EfficientNetB6.yaml | 6 +++--- .../ImageNet/EfficientNet/EfficientNetB7.yaml | 6 +++--- ppcls/configs/ImageNet/ResNeSt/ResNeSt101.yaml | 12 ++++++------ 10 files changed, 34 insertions(+), 34 deletions(-) diff --git a/ppcls/configs/ImageNet/CSPNet/CSPDarkNet53.yaml b/ppcls/configs/ImageNet/CSPNet/CSPDarkNet53.yaml index cce4e3e5..e512e5b3 100644 --- a/ppcls/configs/ImageNet/CSPNet/CSPDarkNet53.yaml +++ b/ppcls/configs/ImageNet/CSPNet/CSPDarkNet53.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 256, 256] save_inference_dir: ./inference # training model under @to_static to_static: False diff --git a/ppcls/configs/ImageNet/DarkNet/DarkNet53.yaml b/ppcls/configs/ImageNet/DarkNet/DarkNet53.yaml index 1a55e75d..ba1478ff 100644 --- a/ppcls/configs/ImageNet/DarkNet/DarkNet53.yaml +++ b/ppcls/configs/ImageNet/DarkNet/DarkNet53.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 256, 256] save_inference_dir: ./inference # model architecture @@ -53,7 +53,7 @@ DataLoader: to_rgb: True channel_first: False - RandCropImage: - size: 224 + size: 256 - RandFlipImage: flip_code: 1 - NormalizeImage: @@ -84,9 +84,9 @@ DataLoader: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 292 - CropImage: - size: 224 + size: 256 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] @@ -109,9 +109,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 292 - CropImage: - size: 224 + size: 256 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB1.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB1.yaml index 502be5ea..92ec91b7 100644 --- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB1.yaml +++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB1.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 240, 240] save_inference_dir: ./inference # model architecture @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 272 - CropImage: - size: 224 + size: 240 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB2.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB2.yaml index 230d5604..16f2e2f1 100644 --- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB2.yaml +++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB2.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 260, 260] save_inference_dir: ./inference # model architecture @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 292 - CropImage: - size: 224 + size: 260 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB3.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB3.yaml index fcd8c013..0847256f 100644 --- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB3.yaml +++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB3.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 300, 300] save_inference_dir: ./inference # model architecture @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 332 - CropImage: - size: 224 + size: 300 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB4.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB4.yaml index 1f97c710..c5396ff2 100644 --- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB4.yaml +++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB4.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 380, 380] save_inference_dir: ./inference # model architecture @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 412 - CropImage: - size: 224 + size: 380 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB5.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB5.yaml index 94656dc8..286971d4 100644 --- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB5.yaml +++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB5.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 456, 456] save_inference_dir: ./inference # model architecture @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 448 - CropImage: - size: 224 + size: 456 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB6.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB6.yaml index 8da4f005..edaab554 100644 --- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB6.yaml +++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB6.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 528, 528] save_inference_dir: ./inference # model architecture @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 560 - CropImage: - size: 224 + size: 528 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml index 2470cff1..1ced43d1 100644 --- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml +++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 600, 600] save_inference_dir: ./inference # model architecture @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 632 - CropImage: - size: 224 + size: 600 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] diff --git a/ppcls/configs/ImageNet/ResNeSt/ResNeSt101.yaml b/ppcls/configs/ImageNet/ResNeSt/ResNeSt101.yaml index 7b3bc2bd..724b56df 100644 --- a/ppcls/configs/ImageNet/ResNeSt/ResNeSt101.yaml +++ b/ppcls/configs/ImageNet/ResNeSt/ResNeSt101.yaml @@ -11,7 +11,7 @@ Global: print_batch_step: 10 use_visualdl: False # used for static mode and model export - image_shape: [3, 224, 224] + image_shape: [3, 256, 256] save_inference_dir: ./inference # model architecture @@ -53,7 +53,7 @@ DataLoader: to_rgb: True channel_first: False - RandCropImage: - size: 224 + size: 256 - RandFlipImage: flip_code: 1 - AutoAugment: @@ -85,9 +85,9 @@ DataLoader: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 288 - CropImage: - size: 224 + size: 256 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] @@ -110,9 +110,9 @@ Infer: to_rgb: True channel_first: False - ResizeImage: - resize_short: 256 + resize_short: 288 - CropImage: - size: 224 + size: 256 - NormalizeImage: scale: 1.0/255.0 mean: [0.485, 0.456, 0.406] -- GitLab