| 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/} | |