33 lines
692 B
YAML
33 lines
692 B
YAML
mode: 'valid'
|
|
ARCHITECTURE:
|
|
name: ""
|
|
|
|
pretrained_model: ""
|
|
classes_num: 1000
|
|
total_images: 1281167
|
|
topk: 5
|
|
image_shape: [3, 224, 224]
|
|
|
|
VALID:
|
|
batch_size: 16
|
|
num_workers: 4
|
|
file_list: "../dataset/ILSVRC2012/val_list.txt"
|
|
data_dir: "../dataset/ILSVRC2012/"
|
|
shuffle_seed: 0
|
|
transforms:
|
|
- DecodeImage:
|
|
to_rgb: True
|
|
to_np: False
|
|
channel_first: False
|
|
- ResizeImage:
|
|
resize_short: 256
|
|
- CropImage:
|
|
size: 224
|
|
- NormalizeImage:
|
|
scale: 1.0/255.0
|
|
mean: [0.485, 0.456, 0.406]
|
|
std: [0.229, 0.224, 0.225]
|
|
order: ''
|
|
- ToCHWImage:
|
|
|