inference_language_classification.yaml 786 字节
Newer Older
1
Global:
2 3
  infer_imgs: "./images/PULC/language_classification/word_35404.png"
  inference_model_dir: "./models/language_classification_infer"
4 5 6 7 8 9 10 11 12 13 14 15 16 17
  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:
18
        size: [160, 80]
19 20 21 22 23 24 25 26 27 28 29 30
    - 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: Topk
  Topk:
    topk: 2
G
gaotingquan 已提交
31
    class_id_map_file: "../ppcls/utils/PULC_label_list/language_classification_label_list.txt"
32 33
  SavePreLabel:
    save_dir: ./pre_label/