mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
Merge pull request #2118 from littletomatodonkey/dev/fix_va_dist
fix cfg
This commit is contained in:
commit
f729688e88
@ -142,6 +142,26 @@ DataLoader:
|
|||||||
num_workers: 8
|
num_workers: 8
|
||||||
use_shared_memory: True
|
use_shared_memory: True
|
||||||
|
|
||||||
|
Infer:
|
||||||
|
infer_imgs: ./deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg
|
||||||
|
batch_size: 10
|
||||||
|
transforms:
|
||||||
|
- DecodeImage:
|
||||||
|
to_rgb: True
|
||||||
|
channel_first: False
|
||||||
|
- ResizeImage:
|
||||||
|
size: [256, 192]
|
||||||
|
- NormalizeImage:
|
||||||
|
scale: 1.0/255.0
|
||||||
|
mean: [0.485, 0.456, 0.406]
|
||||||
|
std: [0.229, 0.224, 0.225]
|
||||||
|
order: ''
|
||||||
|
- ToCHWImage:
|
||||||
|
PostProcess:
|
||||||
|
name: VehicleAttribute
|
||||||
|
color_threshold: 0.5
|
||||||
|
type_threshold: 0.5
|
||||||
|
|
||||||
|
|
||||||
Metric:
|
Metric:
|
||||||
Eval:
|
Eval:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user