tools/accuracy_checker/configs/human-pose-estimation-0001.yml

   1 models:
   2   - name: human-pose-estimation-0001
   3
   4     launchers:
   5       - framework: dlsdk
   6         tags:
   7           - FP32
   8         device: CPU
   9         model:   human-pose-estimation-0001/FP32/human-pose-estimation-0001.xml
  10         weights: human-pose-estimation-0001/FP32/human-pose-estimation-0001.bin
  11         allow_reshape_input: True
  12         adapter:
  13           type: human_pose_estimation
  14           part_affinity_fields_out: Mconv7_stage2_L1
  15           keypoints_heatmap_out: Mconv7_stage2_L2
  16
  17       - framework: dlsdk
  18         tags:
  19           - GPU32
  20         device: GPU
  21         model:   human-pose-estimation-0001/FP32/human-pose-estimation-0001.xml
  22         weights: human-pose-estimation-0001/FP32/human-pose-estimation-0001.bin
  23         allow_reshape_input: True
  24         adapter:
  25           type: human_pose_estimation
  26           part_affinity_fields_out: Mconv7_stage2_L1
  27           keypoints_heatmap_out: Mconv7_stage2_L2
  28
  29       - framework: dlsdk
  30         tags:
  31           - GPU16
  32         device: GPU
  33         model:   human-pose-estimation-0001/FP16/human-pose-estimation-0001.xml
  34         weights: human-pose-estimation-0001/FP16/human-pose-estimation-0001.bin
  35         allow_reshape_input: True
  36         adapter:
  37           type: human_pose_estimation
  38           part_affinity_fields_out: Mconv7_stage2_L1
  39           keypoints_heatmap_out: Mconv7_stage2_L2
  40
  41       - framework: dlsdk
  42         device: MYRIAD
  43         model:   human-pose-estimation-0001/FP16/human-pose-estimation-0001.xml
  44         weights: human-pose-estimation-0001/FP16/human-pose-estimation-0001.bin
  45         allow_reshape_input: True
  46         adapter:
  47           type: human_pose_estimation
  48           part_affinity_fields_out: Mconv7_stage2_L1
  49           keypoints_heatmap_out: Mconv7_stage2_L2
  50
  51       - framework: dlsdk
  52         device: HDDL
  53         model:   human-pose-estimation-0001/FP16/human-pose-estimation-0001.xml
  54         weights: human-pose-estimation-0001/FP16/human-pose-estimation-0001.bin
  55         allow_reshape_input: True
  56         adapter:
  57           type: human_pose_estimation
  58           part_affinity_fields_out: Mconv7_stage2_L1
  59           keypoints_heatmap_out: Mconv7_stage2_L2
  60
  61       - framework: dlsdk
  62         tags:
  63           - FPGA16
  64         device: HETERO:FPGA,CPU
  65         model:   human-pose-estimation-0001/FP32/human-pose-estimation-0001.xml
  66         weights: human-pose-estimation-0001/FP32/human-pose-estimation-0001.bin
  67         allow_reshape_input: True
  68         adapter:
  69           type: human_pose_estimation
  70           part_affinity_fields_out: Mconv7_stage2_L1
  71           keypoints_heatmap_out: Mconv7_stage2_L2
  72         bitstream: 2019R1_A10DK_FP16_ELU.aocx
  73
  74       - framework: dlsdk
  75         tags:
  76           - FPGA11
  77         device: HETERO:FPGA,CPU
  78         model:   human-pose-estimation-0001/FP32/human-pose-estimation-0001.xml
  79         weights: human-pose-estimation-0001/FP32/human-pose-estimation-0001.bin
  80         allow_reshape_input: True
  81         adapter:
  82           type: human_pose_estimation
  83           part_affinity_fields_out: Mconv7_stage2_L1
  84           keypoints_heatmap_out: Mconv7_stage2_L2
  85         bitstream: 2019R1_A10DK_FP11_ELU.aocx
  86
  87
  88     datasets:
  89       - name: ms_coco_keypoints
  90         data_source: val2017
  91         annotation_conversion:
  92           converter: mscoco_keypoints
  93           annotation_file: person_keypoints_val2017.json
  94
  95         preprocessing:
  96           - type: resize
  97             size: 368
  98             interpolation: CUBIC
  99             aspect_ratio_scale: width
 100           - type: padding
 101             stride: 8
 102
 103         postprocessing:
 104           - type: filter
 105             apply_to: annotation
 106             area_range: 1, 10000000000
 107           - type: filter
 108             apply_to: prediction
 109             area_range: 1, 10000000000
 110
 111         metrics:
 112           - name: AP
 113             type: coco_precision
 114             max_detections: 20