mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
36 lines
792 B
YAML
36 lines
792 B
YAML
Global:
|
|
infer_imgs: "images/PULC/table_attribute/val_3253.jpg"
|
|
inference_model_dir: "./models/table_attribute_infer"
|
|
batch_size: 1
|
|
use_gpu: True
|
|
enable_mkldnn: True
|
|
cpu_num_threads: 10
|
|
benchmark: False
|
|
use_fp16: False
|
|
ir_optim: True
|
|
use_tensorrt: False
|
|
gpu_mem: 8000
|
|
enable_profile: False
|
|
|
|
PreProcess:
|
|
transform_ops:
|
|
- ResizeImage:
|
|
size: [224, 224]
|
|
- NormalizeImage:
|
|
scale: 1.0/255.0
|
|
mean: [0.485, 0.456, 0.406]
|
|
std: [0.229, 0.224, 0.225]
|
|
order: ''
|
|
channel_num: 3
|
|
- ToCHWImage:
|
|
|
|
PostProcess:
|
|
main_indicator: TableAttribute
|
|
TableAttribute:
|
|
source_threshold: 0.5
|
|
number_threshold: 0.5
|
|
color_threshold: 0.5
|
|
clarity_threshold : 0.5
|
|
obstruction_threshold: 0.5
|
|
angle_threshold: 0.5
|