diff --git a/deploy/configs/PULC/safety_helmet/inference_safety_helmet.yaml b/deploy/configs/PULC/safety_helmet/inference_safety_helmet.yaml new file mode 100644 index 0000000000000000000000000000000000000000..8a5a6b5c20a5545bcf73ed35e1bd106bf23487eb --- /dev/null +++ b/deploy/configs/PULC/safety_helmet/inference_safety_helmet.yaml @@ -0,0 +1,36 @@ +Global: + infer_imgs: "./images/PULC/safety_helmet/" + inference_model_dir: "./models/safety_helmet_infer" + batch_size: 1 + use_gpu: True + enable_mkldnn: False + cpu_num_threads: 10 + enable_benchmark: True + use_fp16: False + ir_optim: True + use_tensorrt: False + gpu_mem: 8000 + enable_profile: False + +PreProcess: + transform_ops: + - ResizeImage: + resize_short: 256 + - CropImage: + size: 224 + - NormalizeImage: + scale: 0.00392157 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + channel_num: 3 + - ToCHWImage: + +PostProcess: + main_indicator: ThreshOutput + ThreshOutput: + threshold: 0.9 + label_0: wearing + label_1: unwearing + SavePreLabel: + save_dir: ./pre_label/