models/vamc.yaml

dataset:
  path: det_coco_calib
  sampler: {get_data_num: 100, suffix: null}
  transform_ops:
  - {type: DecodeImage}
  - adaptive_side: long
    size: [640, -1]
    type: Resize
  - {pad_to_square: true, type: Pad}
  - {norm_type: null, type: Normalize}
  - {type: ToTensor}
engine:
  add_extra_ops_to_graph:
    params:
      anchor: [3, 11, 7, 7, 6, 17, 14, 12, 9, 24, 25, 20, 32, 32, 71, 27, 81, 75]
      confidence_threshold: 0.25
      nms_threshold: 0.45
      num_classes: 80
      strides: [8, 16, 32]
    type: yolov5_nms
  calibration: {calibrate_mode: kl_divergence, quantize_per_channel: true, stream_mode: true}
  common: {cluster_mode: 0, data_transport_mode: 3}
  dtype: int8
  type: vacc
model:
  checkpoint: best_299.torchscript.pt
  inputs:
    input: [1, 3, 640, 640]
  name: yolov7_tiny_traffic
workspace: {enable_ir: false, save_log: true, work_dir: ./deploy_weights/}