lsf pc update

Author: lsf

README.md

@@ -12,12 +12,20 @@
 
 ## Highlights
 
-- 支持目标检测算法 Yolo 系列 5/X/7/8 ，多目标跟踪算法 Bytetrack、Deepsort；
+- 支持目标检测算法 Yolo 系列 5/X/7/8 ，多目标跟踪算法 Bytetrack；
 - 预处理和后处理实现CUDA核函数，在 jetson 边缘端也能高性能推理；
 - 封装Tensor、Infer，实现内存复用、CPU/GPU 内存之间自动拷贝、引擎上下文管理等，方便使用；
 - 推理过程实现生产者消费者模型，实现预处理和推理的并行化，进一步提升性能；
 - 采用 RAII 思想+接口模式封装应用，使用安全、便捷。
-- 后续即将单目标跟踪算法 STARK、OSTrack。
+
+
+
+## TODO
+
+- [ ]  STARK-lightning
+- [ ]  OSTrack
+- [ ]  yolov6
+- [ ]  yolov8 ptq & qat
 
 
 

workspace/compile_engine.sh

@@ -1,32 +1,32 @@
 echo ""
 echo ""
-echo "************ compile yolov5 models ***************"
-echo ""
-#trtexec --onnx=./onnx_models/yolov5s.onnx \
-#		--saveEngine=./yolov5s.trt \
-#		--buildOnly \
-#		--minShapes=images:1x3x640x640 \
-#		--optShapes=images:1x3x640x640 \
-#		--maxShapes=images:8x3x640x640 \
-#		--fp16
-#
-#echo ""
-#trtexec --onnx=./onnx_models/yolov5s_ptq.onnx \
-#		--saveEngine=./yolov5s_ptq.trt \
-#		--buildOnly \
-#		--minShapes=images:1x3x640x640 \
-#		--optShapes=images:1x3x640x640 \
-#		--maxShapes=images:8x3x640x640 \
-#		--int8
-#
-#echo ""
-#trtexec --onnx=./onnx_models/yolov5s_qat.onnx \
-#		--saveEngine=./yolov5s_qat.trt \
-#		--buildOnly \
-#		--minShapes=images:1x3x640x640 \
-#		--optShapes=images:1x3x640x640 \
-#		--maxShapes=images:8x3x640x640 \
-#		--int8
+echo "************************ compile yolov5 models ***************************"
+echo ""
+trtexec --onnx=./onnx_models/yolov5s.onnx \
+		--saveEngine=./yolov5s.trt \
+		--buildOnly \
+		--minShapes=images:1x3x640x640 \
+		--optShapes=images:1x3x640x640 \
+		--maxShapes=images:8x3x640x640 \
+		--fp16
+
+echo ""
+trtexec --onnx=./onnx_models/yolov5s_ptq.onnx \
+		--saveEngine=./yolov5s_ptq.trt \
+		--buildOnly \
+		--minShapes=images:1x3x640x640 \
+		--optShapes=images:1x3x640x640 \
+		--maxShapes=images:8x3x640x640 \
+		--int8
+
+echo ""
+trtexec --onnx=./onnx_models/yolov5s_qat.onnx \
+		--saveEngine=./yolov5s_qat.trt \
+		--buildOnly \
+		--minShapes=images:1x3x640x640 \
+		--optShapes=images:1x3x640x640 \
+		--maxShapes=images:8x3x640x640 \
+		--int8
 
 trtexec --onnx=./onnx_models/yolov5m.onnx \
 		--saveEngine=./yolov5m.trt \
@@ -48,7 +48,7 @@ trtexec --onnx=./onnx_models/yolov5m_ptq.onnx \
 
 echo ""
 echo ""
-echo "************ compile yolox models ***************"
+echo "************************ compile yolox models ***************************"
 echo ""
 trtexec --onnx=./onnx_models/yolox_s.onnx \
 		--saveEngine=./yolox_s.trt \
@@ -70,7 +70,7 @@ trtexec --onnx=./onnx_models/yolox_m.onnx \
 
 echo ""
 echo ""
-echo "************ compile yolov7 models ***************"
+echo "************************ compile yolov7 models ***************************"
 echo ""
 trtexec --onnx=./onnx_models/yolov7.onnx \
 		--saveEngine=./yolov7.trt \
@@ -92,7 +92,7 @@ trtexec --onnx=./onnx_models/yolov7_qat.onnx \
 
 echo ""
 echo ""
-echo "************ compile yolov8 models ***************"
+echo "************************ compile yolov8 models ***************************"
 echo ""
 trtexec --onnx=./onnx_models/yolov8n.onnx \
 		--saveEngine=./yolov8n.trt \