fix cfg
parent
f3f0605c7e
commit
2796ad1cd7
|
@ -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…
Reference in New Issue