From 283f9fa9b627ebaa2ad95f2e23cfc633aac54603 Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Sat, 18 Sep 2021 03:46:15 +0000 Subject: [PATCH] fix: fix swin training config --- .../SwinTransformer_base_patch4_window12_384.yaml | 6 ++++++ .../SwinTransformer_base_patch4_window7_224.yaml | 6 ++++++ .../SwinTransformer_large_patch4_window12_384.yaml | 6 ++++++ .../SwinTransformer_large_patch4_window7_224.yaml | 6 ++++++ .../SwinTransformer_small_patch4_window7_224.yaml | 6 ++++++ .../SwinTransformer_tiny_patch4_window7_224.yaml | 6 ++++++ 6 files changed, 36 insertions(+) diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml index af54e4aa..5d976c0b 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml @@ -61,6 +61,8 @@ DataLoader: channel_first: False - RandCropImage: size: 384 + interpolation: bicubic + backend: pil - RandFlipImage: flip_code: 1 - TimmAutoAugment: @@ -109,6 +111,8 @@ DataLoader: channel_first: False - ResizeImage: resize_short: 438 + interpolation: bicubic + backend: pil - CropImage: size: 384 - NormalizeImage: @@ -134,6 +138,8 @@ Infer: channel_first: False - ResizeImage: resize_short: 438 + interpolation: bicubic + backend: pil - CropImage: size: 384 - NormalizeImage: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml index 4b9baa1b..efbd427a 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml @@ -61,6 +61,8 @@ DataLoader: channel_first: False - RandCropImage: size: 224 + interpolation: bicubic + backend: pil - RandFlipImage: flip_code: 1 - TimmAutoAugment: @@ -109,6 +111,8 @@ DataLoader: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: @@ -134,6 +138,8 @@ Infer: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml index 58c9667e..6c3abe6f 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml @@ -61,6 +61,8 @@ DataLoader: channel_first: False - RandCropImage: size: 384 + interpolation: bicubic + backend: pil - RandFlipImage: flip_code: 1 - TimmAutoAugment: @@ -109,6 +111,8 @@ DataLoader: channel_first: False - ResizeImage: resize_short: 438 + interpolation: bicubic + backend: pil - CropImage: size: 384 - NormalizeImage: @@ -134,6 +138,8 @@ Infer: channel_first: False - ResizeImage: resize_short: 438 + interpolation: bicubic + backend: pil - CropImage: size: 384 - NormalizeImage: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml index 16f5a7dc..dd2b2acd 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml @@ -61,6 +61,8 @@ DataLoader: channel_first: False - RandCropImage: size: 224 + interpolation: bicubic + backend: pil - RandFlipImage: flip_code: 1 - TimmAutoAugment: @@ -109,6 +111,8 @@ DataLoader: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: @@ -134,6 +138,8 @@ Infer: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml index 88fc3da4..34a80d83 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml @@ -61,6 +61,8 @@ DataLoader: channel_first: False - RandCropImage: size: 224 + interpolation: bicubic + backend: pil - RandFlipImage: flip_code: 1 - TimmAutoAugment: @@ -109,6 +111,8 @@ DataLoader: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: @@ -134,6 +138,8 @@ Infer: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml index ed9b4d50..d9215938 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml @@ -61,6 +61,8 @@ DataLoader: channel_first: False - RandCropImage: size: 224 + interpolation: bicubic + backend: pil - RandFlipImage: flip_code: 1 - TimmAutoAugment: @@ -109,6 +111,8 @@ DataLoader: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: @@ -134,6 +138,8 @@ Infer: channel_first: False - ResizeImage: resize_short: 256 + interpolation: bicubic + backend: pil - CropImage: size: 224 - NormalizeImage: -- GitLab