mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix some configs

pull/1446/head
cuicheng01 2021-11-15 07:26:58 +00:00
parent 6e4bf593fb
commit 423144c097
10 changed files with 34 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/configs/ImageNet/CSPNet/CSPDarkNet53.yaml
View File

@ -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

12
ppcls/configs/ImageNet/DarkNet/DarkNet53.yaml
View File

@ -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]

6
ppcls/configs/ImageNet/EfficientNet/EfficientNetB1.yaml
View File

@ -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]

6
ppcls/configs/ImageNet/EfficientNet/EfficientNetB2.yaml
View File

@ -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]

6
ppcls/configs/ImageNet/EfficientNet/EfficientNetB3.yaml
View File

@ -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]

6
ppcls/configs/ImageNet/EfficientNet/EfficientNetB4.yaml
View File

@ -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]

6
ppcls/configs/ImageNet/EfficientNet/EfficientNetB5.yaml
View File

@ -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]

6
ppcls/configs/ImageNet/EfficientNet/EfficientNetB6.yaml
View File

@ -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]

6
ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml
View File

@ -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]

12
ppcls/configs/ImageNet/ResNeSt/ResNeSt101.yaml
View File

@ -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]