commit
a4711c3934
|
@ -142,6 +142,26 @@ DataLoader:
|
|||
num_workers: 8
|
||||
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:
|
||||
Eval:
|
||||
|
|
Loading…
Reference in New Issue