Skip to content

Commit e3031b0

Browse files
author
lsf
committed
lsf pc update
1 parent f0b1af4 commit e3031b0

File tree

2 files changed

+158
-156
lines changed

2 files changed

+158
-156
lines changed

‎README.md

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@
7979

8080
2. compile engine
8181

82-
1. 下载 onnx 模型 [google driver](https://drive.google.com/drive/folders/16ZqDaxlWm1aDXQsjsxLS7yFL0YqzHbxT?usp=sharing) 或者按照教程导出，教程在各文件夹READEME
82+
1. 下载 onnx 模型 [google driver](https://drive.google.com/drive/folders/16ZqDaxlWm1aDXQsjsxLS7yFL0YqzHbxT?usp=sharing) 或者按照教程导出，教程在各文件夹README
8383

8484
2. ```bash
8585
cd Linfer/workspace
@@ -116,11 +116,13 @@
116116
| yolov7 | **int8** | 640x640 | 49.55 |
117117
| yolov8_n | fp16 | 640x640 | 121.94 |
118118
| yolov8_s | fp16 | 640x640 | 81.40 |
119-
| yolov8_l | fp16 | 640x640 | 13 |
120-
| yolov10_n | fp16 | 640x640 | |
121-
| yolov10_s | fp16 | 640x640 | |
122-
| yolov10_l | fp16 | 640x640 | |
123-
| rtdetr_r50 | fp16 | 640x640 | 12 |
119+
| yolov8_m | fp16 | 640x640 | 41.14 |
120+
| yolov8_l | fp16 | 640x640 | 27.52 |
121+
| yolov10_n | fp16 | 640x640 | 115.13 |
122+
| yolov10_s | fp16 | 640x640 | 73.65 |
123+
| yolov10_m | fp16 | 640x640 | 39.51 |
124+
| yolov10_l | fp16 | 640x640 | 26.41 |
125+
| rtdetr_r50 | fp16 | 640x640 | 11.25 |
124126
| lighttrack | fp16 | 256x256 | 90.91 |
125127
| ostrack | fp16 | 256x256 | 37.04 |
126128
| yolop | fp16 | 640x640 | 31.4 |

‎workspace/compile_engine.sh

Lines changed: 150 additions & 150 deletions
Original file line numberDiff line numberDiff line change
@@ -1,67 +1,67 @@
1-
echo ""
2-
echo ""
3-
echo "************************ compile yolov5 models ***************************"
4-
echo ""
5-
trtexec --onnx=./onnx_models/yolov5s.onnx \
6-
--saveEngine=./yolov5s.trt \
7-
--buildOnly \
8-
--minShapes=images:1x3x640x640 \
9-
--optShapes=images:1x3x640x640 \
10-
--maxShapes=images:8x3x640x640 \
11-
--fp16
12-
13-
echo ""
14-
trtexec --onnx=./onnx_models/yolov5m.onnx \
15-
--saveEngine=./yolov5m.trt \
16-
--buildOnly \
17-
--minShapes=images:1x3x640x640 \
18-
--optShapes=images:1x3x640x640 \
19-
--maxShapes=images:8x3x640x640 \
20-
--fp16
21-
22-
23-
echo ""
24-
echo ""
25-
echo "************************ compile yolox models ***************************"
26-
echo ""
27-
trtexec --onnx=./onnx_models/yolox_s.onnx \
28-
--saveEngine=./yolox_s.trt \
29-
--buildOnly \
30-
--minShapes=images:1x3x640x640 \
31-
--optShapes=images:1x3x640x640 \
32-
--maxShapes=images:8x3x640x640 \
33-
--fp16
34-
35-
echo ""
36-
trtexec --onnx=./onnx_models/yolox_m.onnx \
37-
--saveEngine=./yolox_m.trt \
38-
--buildOnly \
39-
--minShapes=images:1x3x640x640 \
40-
--optShapes=images:1x3x640x640 \
41-
--maxShapes=images:8x3x640x640 \
42-
--fp16
43-
44-
45-
echo ""
46-
echo ""
47-
echo "************************ compile yolov7 models ***************************"
48-
echo ""
49-
trtexec --onnx=./onnx_models/yolov7.onnx \
50-
--saveEngine=./yolov7.trt \
51-
--buildOnly \
52-
--minShapes=images:1x3x640x640 \
53-
--optShapes=images:1x3x640x640 \
54-
--maxShapes=images:8x3x640x640 \
55-
--fp16
56-
57-
echo ""
58-
trtexec --onnx=./onnx_models/yolov7_qat.onnx \
59-
--saveEngine=./yolov7_qat.trt \
60-
--buildOnly \
61-
--minShapes=images:1x3x640x640 \
62-
--optShapes=images:1x3x640x640 \
63-
--maxShapes=images:8x3x640x640 \
64-
--int8
1+
# echo ""
2+
# echo ""
3+
# echo "************************ compile yolov5 models ***************************"
4+
# echo ""
5+
# trtexec --onnx=./onnx_models/yolov5s.onnx \
6+
# --saveEngine=./yolov5s.trt \
7+
# --buildOnly \
8+
# --minShapes=images:1x3x640x640 \
9+
# --optShapes=images:1x3x640x640 \
10+
# --maxShapes=images:8x3x640x640 \
11+
# --fp16
12+
13+
# echo ""
14+
# trtexec --onnx=./onnx_models/yolov5m.onnx \
15+
# --saveEngine=./yolov5m.trt \
16+
# --buildOnly \
17+
# --minShapes=images:1x3x640x640 \
18+
# --optShapes=images:1x3x640x640 \
19+
# --maxShapes=images:8x3x640x640 \
20+
# --fp16
21+
22+
23+
# echo ""
24+
# echo ""
25+
# echo "************************ compile yolox models ***************************"
26+
# echo ""
27+
# trtexec --onnx=./onnx_models/yolox_s.onnx \
28+
# --saveEngine=./yolox_s.trt \
29+
# --buildOnly \
30+
# --minShapes=images:1x3x640x640 \
31+
# --optShapes=images:1x3x640x640 \
32+
# --maxShapes=images:8x3x640x640 \
33+
# --fp16
34+
35+
# echo ""
36+
# trtexec --onnx=./onnx_models/yolox_m.onnx \
37+
# --saveEngine=./yolox_m.trt \
38+
# --buildOnly \
39+
# --minShapes=images:1x3x640x640 \
40+
# --optShapes=images:1x3x640x640 \
41+
# --maxShapes=images:8x3x640x640 \
42+
# --fp16
43+
44+
45+
# echo ""
46+
# echo ""
47+
# echo "************************ compile yolov7 models ***************************"
48+
# echo ""
49+
# trtexec --onnx=./onnx_models/yolov7.onnx \
50+
# --saveEngine=./yolov7.trt \
51+
# --buildOnly \
52+
# --minShapes=images:1x3x640x640 \
53+
# --optShapes=images:1x3x640x640 \
54+
# --maxShapes=images:8x3x640x640 \
55+
# --fp16
56+
57+
# echo ""
58+
# trtexec --onnx=./onnx_models/yolov7_qat.onnx \
59+
# --saveEngine=./yolov7_qat.trt \
60+
# --buildOnly \
61+
# --minShapes=images:1x3x640x640 \
62+
# --optShapes=images:1x3x640x640 \
63+
# --maxShapes=images:8x3x640x640 \
64+
# --int8
6565

6666

6767
echo ""
@@ -144,89 +144,89 @@ trtexec --onnx=./onnx_models/yolov10l.onnx \
144144
--fp16
145145

146146

147-
echo ""
148-
echo ""
149-
echo "************************ compile RT-DETR models ***************************"
150-
echo ""
151-
trtexec --onnx=./onnx_models/rtdetr_r50vd_6x_coco_dynamic.onnx \
152-
--saveEngine=./rtdetr_r50vd_6x_coco_dynamic_fp16.trt \
153-
--buildOnly \
154-
--minShapes=image:1x3x640x640 \
155-
--optShapes=image:1x3x640x640 \
156-
--maxShapes=image:8x3x640x640 \
157-
--fp16
158-
159-
160-
echo ""
161-
echo ""
162-
echo "************************ compile LightTrack models ***************************"
163-
echo ""
164-
trtexec --onnx=./onnx_models/lighttrack-z.onnx \
165-
--saveEngine=./lighttrack-z.trt \
166-
--buildOnly \
167-
--fp16
168-
169-
echo ""
170-
trtexec --onnx=./onnx_models/lighttrack-x-head.onnx \
171-
--saveEngine=./lighttrack-x-head.trt \
172-
--buildOnly \
173-
--fp16
174-
175-
176-
echo ""
177-
echo ""
178-
echo "************************ compile OSTrack models ***************************"
179-
echo ""
180-
trtexec --onnx=./onnx_models/ostrack-256.onnx \
181-
--saveEngine=./ostrack-256.trt \
182-
--buildOnly \
183-
--fp16
184-
185-
echo ""
186-
trtexec --onnx=./onnx_models/ostrack-384-ce.onnx \
187-
--saveEngine=./ostrack-384-ce.trt \
188-
--buildOnly \
189-
--fp16
190-
191-
192-
echo ""
193-
echo ""
194-
echo "************************ compile YoloP model ***************************"
195-
echo ""
196-
trtexec --onnx=./onnx_models/yolop-640.onnx \
197-
--saveEngine=./yolop-640.trt \
198-
--buildOnly \
199-
--fp16
200-
201-
echo ""
202-
trtexec --onnx=./onnx_models/yolopv2-480x640.onnx \
203-
--saveEngine=./yolopv2-480x640.trt \
204-
--buildOnly \
205-
--fp16
206-
207-
208-
echo ""
209-
echo ""
210-
echo "************************ compile PPSeg model ***************************"
211-
echo ""
212-
trtexec --onnx=./onnx_models/mobileseg_mbn3.onnx \
213-
--saveEngine=./mobileseg_mbn3.trt \
214-
--buildOnly \
215-
--fp16
216-
217-
echo ""
218-
trtexec --onnx=./onnx_models/ppliteseg_stdc2.onnx \
219-
--saveEngine=./ppliteseg_stdc2.trt \
220-
--buildOnly \
221-
--fp16
222-
223-
224-
225-
echo ""
226-
echo ""
227-
echo "************************ compile MonoDepth model ***************************"
228-
echo ""
229-
trtexec --onnx=./onnx_models/litemono-b.onnx \
230-
--saveEngine=./litemono-b.trt \
231-
--buildOnly \
232-
--fp16
147+
# echo ""
148+
# echo ""
149+
# echo "************************ compile RT-DETR models ***************************"
150+
# echo ""
151+
# trtexec --onnx=./onnx_models/rtdetr_r50vd_6x_coco_dynamic.onnx \
152+
# --saveEngine=./rtdetr_r50vd_6x_coco_dynamic_fp16.trt \
153+
# --buildOnly \
154+
# --minShapes=image:1x3x640x640 \
155+
# --optShapes=image:1x3x640x640 \
156+
# --maxShapes=image:8x3x640x640 \
157+
# --fp16
158+
159+
160+
# echo ""
161+
# echo ""
162+
# echo "************************ compile LightTrack models ***************************"
163+
# echo ""
164+
# trtexec --onnx=./onnx_models/lighttrack-z.onnx \
165+
# --saveEngine=./lighttrack-z.trt \
166+
# --buildOnly \
167+
# --fp16
168+
169+
# echo ""
170+
# trtexec --onnx=./onnx_models/lighttrack-x-head.onnx \
171+
# --saveEngine=./lighttrack-x-head.trt \
172+
# --buildOnly \
173+
# --fp16
174+
175+
176+
# echo ""
177+
# echo ""
178+
# echo "************************ compile OSTrack models ***************************"
179+
# echo ""
180+
# trtexec --onnx=./onnx_models/ostrack-256.onnx \
181+
# --saveEngine=./ostrack-256.trt \
182+
# --buildOnly \
183+
# --fp16
184+
185+
# echo ""
186+
# trtexec --onnx=./onnx_models/ostrack-384-ce.onnx \
187+
# --saveEngine=./ostrack-384-ce.trt \
188+
# --buildOnly \
189+
# --fp16
190+
191+
192+
# echo ""
193+
# echo ""
194+
# echo "************************ compile YoloP model ***************************"
195+
# echo ""
196+
# trtexec --onnx=./onnx_models/yolop-640.onnx \
197+
# --saveEngine=./yolop-640.trt \
198+
# --buildOnly \
199+
# --fp16
200+
201+
# echo ""
202+
# trtexec --onnx=./onnx_models/yolopv2-480x640.onnx \
203+
# --saveEngine=./yolopv2-480x640.trt \
204+
# --buildOnly \
205+
# --fp16
206+
207+
208+
# echo ""
209+
# echo ""
210+
# echo "************************ compile PPSeg model ***************************"
211+
# echo ""
212+
# trtexec --onnx=./onnx_models/mobileseg_mbn3.onnx \
213+
# --saveEngine=./mobileseg_mbn3.trt \
214+
# --buildOnly \
215+
# --fp16
216+
217+
# echo ""
218+
# trtexec --onnx=./onnx_models/ppliteseg_stdc2.onnx \
219+
# --saveEngine=./ppliteseg_stdc2.trt \
220+
# --buildOnly \
221+
# --fp16
222+
223+
224+
225+
# echo ""
226+
# echo ""
227+
# echo "************************ compile MonoDepth model ***************************"
228+
# echo ""
229+
# trtexec --onnx=./onnx_models/litemono-b.onnx \
230+
# --saveEngine=./litemono-b.trt \
231+
# --buildOnly \
232+
# --fp16

0 commit comments

Comments
 (0)