vastaisz commited on
Commit
d16692d
·
verified ·
1 Parent(s): c34c9ac

Upload detection_track model

Browse files
models/detection_track/yolov5s_3_1024_1024-int8-kl_divergence-1_3_1024_1024-vacc-pipeline/mod.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e661f1380d506dae9d00d8585859da00532b2cb4fa3b0006a0d3a8eafa73f25
3
+ size 884173
models/detection_track/yolov5s_3_1024_1024-int8-kl_divergence-1_3_1024_1024-vacc-pipeline/mod.params ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:799a85b027b1ebe18e7b638834ade4ff1b2132e6071d693f140de9ef598b4aa4
3
+ size 9498738
models/detection_track/yolov5s_3_1024_1024-int8-kl_divergence-1_3_1024_1024-vacc-pipeline/mod.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9890572ded3f4eb61de1f1c7adfa2dea8a151070bd693cdc37edcb2a29bb86ef
3
+ size 931264
models/detection_track/yolov5s_3_1024_1024-int8-kl_divergence-1_3_1024_1024-vacc-pipeline/vamc.yaml ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset:
2
+ path: widerface_quant_images_305_calib
3
+ sampler: {get_data_num: 50, suffix: jpg}
4
+ transform_ops:
5
+ - {type: DecodeImage}
6
+ - adaptive_side: long
7
+ size: [1024, -1]
8
+ type: Resize
9
+ - {pad_to_square: true, type: Pad}
10
+ - {div255: true, norm_type: null, type: Normalize}
11
+ - {type: ToTensor}
12
+ engine:
13
+ add_extra_ops_to_graph:
14
+ params:
15
+ anchor: [10, 13, 16, 30, 33, 23, 30, 61, 62, 45, 59, 119, 116, 90, 156, 198,
16
+ 373, 326]
17
+ confidence_threshold: 0.25
18
+ nms_threshold: 0.45
19
+ num_classes: 1
20
+ strides: [8, 16, 32]
21
+ type: yolov5_nms
22
+ calibration: {calibrate_mode: kl_divergence, quantize_per_channel: true, stream_mode: true}
23
+ common: {cluster_mode: 0, data_transport_mode: 3}
24
+ dtype: int8
25
+ type: vacc
26
+ model:
27
+ checkpoint: yolov5s_3_1024_1024.torchscript.pt
28
+ inputs:
29
+ images: [1, 3, 1024, 1024]
30
+ name: yolov5s_3_1024_1024
31
+ workspace: {enable_ir: false, save_log: true, work_dir: ./deploy_weights/}