build_product.yaml 814 字节
Newer Older
C
cuicheng01 已提交
1
Global:
C
cuicheng01 已提交
2
  rec_inference_model_dir: "./models/product_ResNet50_vd_aliproduct_v1.0_infer"
C
cuicheng01 已提交
3 4
  batch_size: 1
  use_gpu: True
W
weishengyu 已提交
5
  enable_mkldnn: False
C
cuicheng01 已提交
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
  cpu_num_threads: 100
  enable_benchmark: True
  use_fp16: False
  ir_optim: True
  use_tensorrt: False
  gpu_mem: 8000
  enable_profile: False

RecPreProcess:
  transform_ops:
    - ResizeImage:
        size: 224
    - NormalizeImage:
        scale: 0.00392157
        mean: [0.485, 0.456, 0.406]
        std: [0.229, 0.224, 0.225]
        order: ''
    - ToCHWImage:

RecPostProcess: null

# indexing engine config
IndexProcess:
C
cuicheng01 已提交
29 30 31
  index_path: "./dataset/product_demo_data_v1.0/index"
  image_root: "./dataset/product_demo_data_v1.0"
  data_file:  "./dataset/product_demo_data_v1.0/data_file.txt"
C
cuicheng01 已提交
32 33 34 35
  delimiter: " "
  dist_type: "IP"
  pq_size: 100
  embedding_size: 512