Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
接近 2 年 前同步成功

通知 116
Star 4999
Fork 1114
P
PaddleClas
提交 1ca24932 编写于 作者: C cuicheng01
浏览文件
操作

Update configs

上级 4afc61a3
隐藏空白更改
内联 并排
Showing with 1629 addition and 1627 deletion
ppcls/configs/ImageNet/HRNet/HRNet_W18_C.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
...@@ -46,80 +46,80 @@ Optimizer: ...@@ -46,80 +46,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/HRNet/HRNet_W30_C.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
...@@ -46,80 +46,80 @@ Optimizer: ...@@ -46,80 +46,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/HRNet/HRNet_W32_C.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
...@@ -46,80 +46,80 @@ Optimizer: ...@@ -46,80 +46,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/HRNet/HRNet_W40_C.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
...@@ -46,80 +46,80 @@ Optimizer: ...@@ -46,80 +46,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/HRNet/HRNet_W44_C.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
...@@ -46,78 +46,80 @@ Optimizer: ...@@ -46,78 +46,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
size: 224 resize_short: 256
- NormalizeImage: - CropImage:
scale: 0.00392157 size: 224
mean: [0.485, 0.456, 0.406] - NormalizeImage:
std: [0.229, 0.224, 0.225] scale: 1.0/255.0
order: '' mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/HRNet/HRNet_W48_C.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
...@@ -46,80 +46,80 @@ Optimizer: ...@@ -46,80 +46,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/HRNet/HRNet_W64_C.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
...@@ -46,80 +46,80 @@ Optimizer: ...@@ -46,80 +46,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/Ineption/InceptionV3.yaml
浏览文件 @ 1ca24932
...@@ -12,7 +12,7 @@ Global: ...@@ -12,7 +12,7 @@ Global:
print_batch_step: 10 print_batch_step: 10
use_visualdl: False use_visualdl: False
# used for static mode and model export # used for static mode and model export
image_shape: [3, 224, 224] image_shape: [3, 299, 299]
save_inference_dir: "./inference" save_inference_dir: "./inference"
# model architecture # model architecture
...@@ -24,19 +24,18 @@ Loss: ...@@ -24,19 +24,18 @@ Loss:
Train: Train:
- CELoss: - CELoss:
weight: 1.0 weight: 1.0
epsilon: 0.1
Eval: Eval:
- CELoss: - CELoss:
weight: 1.0 weight: 1.0
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 0.045 learning_rate: 0.045
decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001]
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.0001 coeff: 0.0001
...@@ -46,80 +45,80 @@ Optimizer: ...@@ -46,80 +45,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 299 size: 299
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 320 resize_short: 320
- CropImage: - CropImage:
size: 299 size: 299
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 320 resize_short: 320
- CropImage: - CropImage:
size: 299 size: 299
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml
浏览文件 @ 1ca24932
...@@ -30,96 +30,96 @@ Loss: ...@@ -30,96 +30,96 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.00003 coeff: 0.0003
# data loader for train and eval # data loader for train and eval
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_25.yaml
浏览文件 @ 1ca24932
...@@ -30,96 +30,96 @@ Loss: ...@@ -30,96 +30,96 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.00003 coeff: 0.0003
# data loader for train and eval # data loader for train and eval
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_5.yaml
浏览文件 @ 1ca24932
...@@ -30,96 +30,96 @@ Loss: ...@@ -30,96 +30,96 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.00003 coeff: 0.0003
# data loader for train and eval # data loader for train and eval
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_75.yaml
浏览文件 @ 1ca24932
...@@ -30,96 +30,96 @@ Loss: ...@@ -30,96 +30,96 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Piecewise name: "Piecewise"
learning_rate: 0.1 learning_rate: 0.1
decay_epochs: [30, 60, 90] decay_epochs: [30, 60, 90]
values: [0.1, 0.01, 0.001, 0.0001] values: [0.1, 0.01, 0.001, 0.0001]
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.00003 coeff: 0.0003
# data loader for train and eval # data loader for train and eval
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x0_35.yaml
浏览文件 @ 1ca24932
...@@ -31,10 +31,10 @@ Loss: ...@@ -31,10 +31,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -45,80 +45,80 @@ Optimizer: ...@@ -45,80 +45,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x0_5.yaml
浏览文件 @ 1ca24932
...@@ -31,10 +31,10 @@ Loss: ...@@ -31,10 +31,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -45,80 +45,80 @@ Optimizer: ...@@ -45,80 +45,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x0_75.yaml
浏览文件 @ 1ca24932
...@@ -31,10 +31,10 @@ Loss: ...@@ -31,10 +31,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -45,80 +45,80 @@ Optimizer: ...@@ -45,80 +45,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_0.yaml
浏览文件 @ 1ca24932
...@@ -31,10 +31,10 @@ Loss: ...@@ -31,10 +31,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -45,81 +45,81 @@ Optimizer: ...@@ -45,81 +45,81 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- AutoAugment: - AutoAugment:
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_large_x1_25.yaml
浏览文件 @ 1ca24932
...@@ -31,94 +31,94 @@ Loss: ...@@ -31,94 +31,94 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.00002 coeff: 0.00004
# data loader for train and eval # data loader for train and eval
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x0_35.yaml
浏览文件 @ 1ca24932
...@@ -31,94 +31,94 @@ Loss: ...@@ -31,94 +31,94 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.00002 coeff: 0.00001
# data loader for train and eval # data loader for train and eval
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x0_5.yaml
浏览文件 @ 1ca24932
...@@ -31,94 +31,94 @@ Loss: ...@@ -31,94 +31,94 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
coeff: 0.00002 coeff: 0.00001
# data loader for train and eval # data loader for train and eval
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x0_75.yaml
浏览文件 @ 1ca24932
...@@ -31,10 +31,10 @@ Loss: ...@@ -31,10 +31,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -45,80 +45,80 @@ Optimizer: ...@@ -45,80 +45,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0.yaml
浏览文件 @ 1ca24932
...@@ -31,10 +31,10 @@ Loss: ...@@ -31,10 +31,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -45,80 +45,80 @@ Optimizer: ...@@ -45,80 +45,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_25.yaml
浏览文件 @ 1ca24932
...@@ -31,10 +31,10 @@ Loss: ...@@ -31,10 +31,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 1.3 learning_rate: 1.3
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -45,80 +45,80 @@ Optimizer: ...@@ -45,80 +45,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 512 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/ResNet/ResNet101.yaml
浏览文件 @ 1ca24932
...@@ -66,7 +66,7 @@ DataLoader: ...@@ -66,7 +66,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -91,7 +91,7 @@ DataLoader: ...@@ -91,7 +91,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet101_vd.yaml
浏览文件 @ 1ca24932
...@@ -68,7 +68,7 @@ DataLoader: ...@@ -68,7 +68,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -93,7 +93,7 @@ DataLoader: ...@@ -93,7 +93,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet152.yaml
浏览文件 @ 1ca24932
...@@ -66,7 +66,7 @@ DataLoader: ...@@ -66,7 +66,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -91,7 +91,7 @@ DataLoader: ...@@ -91,7 +91,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet152_vd.yaml
浏览文件 @ 1ca24932
...@@ -68,7 +68,7 @@ DataLoader: ...@@ -68,7 +68,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -93,7 +93,7 @@ DataLoader: ...@@ -93,7 +93,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet18.yaml
浏览文件 @ 1ca24932
...@@ -66,7 +66,7 @@ DataLoader: ...@@ -66,7 +66,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -91,7 +91,7 @@ DataLoader: ...@@ -91,7 +91,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet18_vd.yaml
浏览文件 @ 1ca24932
...@@ -68,7 +68,7 @@ DataLoader: ...@@ -68,7 +68,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -93,7 +93,7 @@ DataLoader: ...@@ -93,7 +93,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet200_vd.yaml
浏览文件 @ 1ca24932
...@@ -68,7 +68,7 @@ DataLoader: ...@@ -68,7 +68,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -93,7 +93,7 @@ DataLoader: ...@@ -93,7 +93,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet34.yaml
浏览文件 @ 1ca24932
...@@ -66,7 +66,7 @@ DataLoader: ...@@ -66,7 +66,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -91,7 +91,7 @@ DataLoader: ...@@ -91,7 +91,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet34_vd.yaml
浏览文件 @ 1ca24932
...@@ -68,7 +68,7 @@ DataLoader: ...@@ -68,7 +68,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -93,7 +93,7 @@ DataLoader: ...@@ -93,7 +93,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet50.yaml
浏览文件 @ 1ca24932
...@@ -66,7 +66,7 @@ DataLoader: ...@@ -66,7 +66,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -91,7 +91,7 @@ DataLoader: ...@@ -91,7 +91,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/ResNet/ResNet50_vd.yaml
浏览文件 @ 1ca24932
...@@ -68,7 +68,7 @@ DataLoader: ...@@ -68,7 +68,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
...@@ -93,7 +93,7 @@ DataLoader: ...@@ -93,7 +93,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
......
ppcls/configs/ImageNet/VGG/VGG11.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 0.1 learning_rate: 0.1
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -44,80 +44,80 @@ Optimizer: ...@@ -44,80 +44,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 128
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/VGG/VGG13.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 0.01 learning_rate: 0.01
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -44,80 +44,80 @@ Optimizer: ...@@ -44,80 +44,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/VGG/VGG16.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 0.01 learning_rate: 0.01
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -44,80 +44,80 @@ Optimizer: ...@@ -44,80 +44,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/ImageNet/VGG/VGG19.yaml
浏览文件 @ 1ca24932
...@@ -30,10 +30,10 @@ Loss: ...@@ -30,10 +30,10 @@ Loss:
Optimizer: Optimizer:
name: Momentum name: "Momentum"
momentum: 0.9 momentum: 0.9
lr: lr:
name: Cosine name: "Cosine"
learning_rate: 0.01 learning_rate: 0.01
regularizer: regularizer:
name: 'L2' name: 'L2'
...@@ -44,80 +44,80 @@ Optimizer: ...@@ -44,80 +44,80 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/train_list.txt" cls_label_path: "./dataset/ILSVRC2012/train_list.txt"
transform_ops: transform_ops:
- RandCropImage: - RandCropImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: ImageNetDataset name: "ImageNetDataset"
image_root: "./dataset/ILSVRC2012/" image_root: "./dataset/ILSVRC2012/"
cls_label_path: "./dataset/ILSVRC2012/val_list.txt" cls_label_path: "./dataset/ILSVRC2012/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Infer: Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
PostProcess: PostProcess:
name: Topk name: "Topk"
topk: 5 topk: 5
class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt" class_id_map_file: "ppcls/utils/imagenet1k_label_list.txt"
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
Eval: Eval:
- TopkAcc: - TopkAcc:
topk: [1, 5] topk: [1, 5]
ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml
浏览文件 @ 1ca24932
...@@ -21,7 +21,7 @@ Arch: ...@@ -21,7 +21,7 @@ Arch:
name: "RecModel" name: "RecModel"
Backbone: Backbone:
name: "ResNet50_vd" name: "ResNet50_vd"
pretrained: False pretrained: True
BackboneStopLayer: BackboneStopLayer:
name: "flatten_0" name: "flatten_0"
Neck: Neck:
...@@ -56,52 +56,52 @@ Optimizer: ...@@ -56,52 +56,52 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Aliproduct/" image_root: "./dataset/Aliproduct/"
cls_label_path: "./dataset/Aliproduct/train_list.txt" cls_label_path: "./dataset/Aliproduct/train_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 0.00392157
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Aliproduct/" image_root: "./dataset/Aliproduct/"
cls_label_path: "./dataset/Aliproduct/val_list.txt" cls_label_path: "./dataset/Aliproduct/val_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
resize_short: 256 resize_short: 256
- CropImage: - CropImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 0.00392157
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
...@@ -114,14 +114,14 @@ Infer: ...@@ -114,14 +114,14 @@ Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 224 resize_short: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
ppcls/configs/Products/ResNet50_vd_Inshop.yaml
浏览文件 @ 1ca24932
# global configs # global configs
Global: Global:
checkpoints: null checkpoints: null
pretrained_model: null # please download pretrained model via this link:
# https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/product_ResNet50_vd_Aliproduct_v1.0_pretrained.pdparams
pretrained_model: "product_ResNet50_vd_Aliproduct_v1.0_pretrained"
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
class_num: 3997 class_num: 3997
...@@ -19,6 +21,9 @@ Global: ...@@ -19,6 +21,9 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: "RecModel" name: "RecModel"
infer_output_key: "features"
infer_add_softmax: False
Backbone: Backbone:
name: "ResNet50_vd" name: "ResNet50_vd"
pretrained: False pretrained: False
...@@ -34,8 +39,6 @@ Arch: ...@@ -34,8 +39,6 @@ Arch:
class_num: 3997 class_num: 3997
margin: 0.15 margin: 0.15
scale: 30 scale: 30
infer_output_key: "features"
infer_add_softmax: False
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
...@@ -67,38 +70,37 @@ Optimizer: ...@@ -67,38 +70,37 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Inshop/" image_root: "./dataset/Inshop/"
cls_label_path: "./dataset/Inshop/train_list.txt" cls_label_path: "./dataset/Inshop/train_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 0.00392157
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- RandomErasing: - RandomErasing:
EPSILON: 0.5 EPSILON: 0.5
sl: 0.02 sl: 0.02
sh: 0.4 sh: 0.4
r1: 0.3 r1: 0.3
mean: [0., 0., 0.] mean: [0., 0., 0.]
sampler: sampler:
name: DistributedRandomIdentitySampler name: "DistributedRandomIdentitySampler"
batch_size: 64 batch_size: 64
num_instances: 2 num_instances: 2
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Eval: Eval:
Query: Query:
# TOTO: modify to the latest trainer
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Inshop/" image_root: "./dataset/Inshop/"
...@@ -112,16 +114,15 @@ DataLoader: ...@@ -112,16 +114,15 @@ DataLoader:
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Gallery: Gallery:
# TOTO: modify to the latest trainer
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Inshop/" image_root: "./dataset/Inshop/"
...@@ -130,17 +131,17 @@ DataLoader: ...@@ -130,17 +131,17 @@ DataLoader:
- ResizeImage: - ResizeImage:
size: 224 size: 224
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
sampler: sampler:
name: DistributedBatchSampler name: "DistributedBatchSampler"
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Metric: Metric:
...@@ -152,14 +153,14 @@ Infer: ...@@ -152,14 +153,14 @@ Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 224 resize_short: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
ppcls/configs/Products/ResNet50_vd_SOP.yaml
浏览文件 @ 1ca24932
# global configs # global configs
Global: Global:
checkpoints: null checkpoints: null
pretrained_model: null # please download pretrained model via this link:
# https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/product_ResNet50_vd_Aliproduct_v1.0_pretrained.pdparams
pretrained_model: "product_ResNet50_vd_Aliproduct_v1.0_pretrained"
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
class_num: 11319 class_num: 11319
...@@ -67,38 +69,37 @@ Optimizer: ...@@ -67,38 +69,37 @@ Optimizer:
DataLoader: DataLoader:
Train: Train:
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Stanford_Online_Products/" image_root: "./dataset/Stanford_Online_Products/"
cls_label_path: "./dataset/Stanford_Online_Products/train_list.txt" cls_label_path: "./dataset/Stanford_Online_Products/train_list.txt"
transform_ops: transform_ops:
- ResizeImage: - ResizeImage:
size: 224 size: 224
- RandFlipImage: - RandFlipImage:
flip_code: 1 flip_code: 1
- NormalizeImage: - NormalizeImage:
scale: 0.00392157 scale: 0.00392157
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- RandomErasing: - RandomErasing:
EPSILON: 0.5 EPSILON: 0.5
sl: 0.02 sl: 0.02
sh: 0.4 sh: 0.4
r1: 0.3 r1: 0.3
mean: [0., 0., 0.] mean: [0., 0., 0.]
sampler: sampler:
name: DistributedRandomIdentitySampler name: DistributedRandomIdentitySampler
batch_size: 64 batch_size: 64
num_instances: 2 num_instances: 2
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: True use_shared_memory: True
Eval: Eval:
Query: Query:
# TOTO: modify to the latest trainer
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Stanford_Online_Products/" image_root: "./dataset/Stanford_Online_Products/"
...@@ -117,11 +118,10 @@ DataLoader: ...@@ -117,11 +118,10 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Gallery: Gallery:
# TOTO: modify to the latest trainer
dataset: dataset:
name: "ImageNetDataset" name: "ImageNetDataset"
image_root: "./dataset/Stanford_Online_Products/" image_root: "./dataset/Stanford_Online_Products/"
...@@ -140,7 +140,7 @@ DataLoader: ...@@ -140,7 +140,7 @@ DataLoader:
drop_last: False drop_last: False
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 4
use_shared_memory: True use_shared_memory: True
Metric: Metric:
...@@ -152,14 +152,14 @@ Infer: ...@@ -152,14 +152,14 @@ Infer:
infer_imgs: "docs/images/whl/demo.jpg" infer_imgs: "docs/images/whl/demo.jpg"
batch_size: 10 batch_size: 10
transforms: transforms:
- DecodeImage: - DecodeImage:
to_rgb: True to_rgb: True
channel_first: False channel_first: False
- ResizeImage: - ResizeImage:
resize_short: 224 resize_short: 224
- NormalizeImage: - NormalizeImage:
scale: 1.0/255.0 scale: 1.0/255.0
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
order: '' order: ''
- ToCHWImage: - ToCHWImage:
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或