vastaisz commited on
Commit
9e6a00e
·
verified ·
1 Parent(s): 4129603

Upload traffic structure configuration

Browse files
Files changed (4) hide show
  1. models/mod.json +2 -2
  2. models/mod.params +2 -2
  3. models/mod.so +2 -2
  4. models/vamc.yaml +19 -14
models/mod.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f7281c513b89a9b1234b9f80c2e50bcae139a134df0cec4b54edfb110702ebe
3
- size 297963
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9eca932da83f52a228d477984390980e68f77e72d9d61366608d72742f940594
3
+ size 730115
models/mod.params CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d0db981294f07ddb42317690e5a71cb4cde3b24800e0708445662b4cb4256494
3
- size 15126980
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:995f587ca934a2593c029f338311e2bbe8d1394d7417e078539a5540598fb7f2
3
+ size 7718214
models/mod.so CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:05b51410019d80450df8c80f881f19152334862f9883479e80f25281a98609f9
3
- size 497408
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dae70418f96679dca742bcb345855a1fb8ed8773835d89bfaf53dff01eec6aa5
3
+ size 852336
models/vamc.yaml CHANGED
@@ -1,25 +1,30 @@
1
  dataset:
2
- path: ILSVRC2012_img_calib
3
- sampler: {get_data_num: 1000, suffix: JPEG}
4
  transform_ops:
5
  - {type: DecodeImage}
6
- - size: [256, 256]
 
7
  type: Resize
8
- - crop_size: [224, 224]
9
- type: CenterCrop
10
- - mean: [0.485, 0.456, 0.406]
11
- std: [0.229, 0.224, 0.225]
12
- type: Normalize
13
  - {type: ToTensor}
14
  engine:
15
- add_extra_ops_to_graph: {type: softmax}
16
- calibration: {calibrate_mode: percentile, quantize_per_channel: true, stream_mode: true}
17
- common: {cluster_mode: 0}
 
 
 
 
 
 
 
18
  dtype: int8
19
  type: vacc
20
  model:
21
- checkpoint: onnx/nmv_type.onnx
22
  inputs:
23
- input: [1, 3, 224, 224]
24
- name: nmv_type
25
  workspace: {enable_ir: false, save_log: true, work_dir: ./deploy_weights/}
 
1
  dataset:
2
+ path: det_coco_calib
3
+ sampler: {get_data_num: 100, suffix: null}
4
  transform_ops:
5
  - {type: DecodeImage}
6
+ - adaptive_side: long
7
+ size: [640, -1]
8
  type: Resize
9
+ - {pad_to_square: true, type: Pad}
10
+ - {norm_type: null, type: Normalize}
 
 
 
11
  - {type: ToTensor}
12
  engine:
13
+ add_extra_ops_to_graph:
14
+ params:
15
+ anchor: [3, 11, 7, 7, 6, 17, 14, 12, 9, 24, 25, 20, 32, 32, 71, 27, 81, 75]
16
+ confidence_threshold: 0.25
17
+ nms_threshold: 0.45
18
+ num_classes: 80
19
+ strides: [8, 16, 32]
20
+ type: yolov5_nms
21
+ calibration: {calibrate_mode: kl_divergence, quantize_per_channel: true, stream_mode: true}
22
+ common: {cluster_mode: 0, data_transport_mode: 3}
23
  dtype: int8
24
  type: vacc
25
  model:
26
+ checkpoint: best_299.torchscript.pt
27
  inputs:
28
+ input: [1, 3, 640, 640]
29
+ name: yolov7_tiny_traffic
30
  workspace: {enable_ir: false, save_log: true, work_dir: ./deploy_weights/}