test_config.yml 830 字节
Newer Older
W
weishengyu 已提交
1 2 3 4 5 6 7
Modules:
  - name: Detector
    type: AlgoMod
    processors:
      - name: ImageProcessor
        type: preprocessor
        ops:
G
gaotingquan 已提交
8 9 10 11 12 13 14 15 16 17
          - ResizeImage:
              size: [640, 640]
              interpolation: 2
          - NormalizeImage:
              scale: 0.00392157
              mean: [0.485, 0.456, 0.406]
              std: [0.229, 0.224, 0.225]
              order: hwc
          - ToCHWImage:
          - GetShapeInfo:
W
weishengyu 已提交
18 19
              configs:
                order: chw
G
gaotingquan 已提交
20
          - ToBatch:
W
weishengyu 已提交
21 22
      - name: PaddlePredictor
        type: predictor
G
gaotingquan 已提交
23
        inference_model_dir: ./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer/
W
weishengyu 已提交
24
        input_names:
G
gaotingquan 已提交
25
        output_names:
W
weishengyu 已提交
26 27
      - name: PPYOLOv2PostPro
        type: postprocessor
G
gaotingquan 已提交
28 29 30 31
        threshold: 0.2
        max_det_results: 1
        label_list:
        - foreground