解决小目标检测问题的算法集

chenlai 34896b8f6d 训练代码地址 1 mēnesi atpakaļ
CMakeFiles 2aa98f254c init 3 mēneši atpakaļ
onnxs 3ffe2db4d5 new onnx models 1 mēnesi atpakaļ
.gitignore 8bce4de154 Initial commit 3 mēneši atpakaļ
CMakeCache.txt 2aa98f254c init 3 mēneši atpakaļ
CMakeLists.txt 2aa98f254c init 3 mēneši atpakaļ
Makefile 2aa98f254c init 3 mēneši atpakaļ
README.md 34896b8f6d 训练代码地址 1 mēnesi atpakaļ
Snipaste_2024-11-26_16-11-41.png 3ffe2db4d5 new onnx models 1 mēnesi atpakaļ
cmake_install.cmake 2aa98f254c init 3 mēneši atpakaļ
create_trt.cpp 2aa98f254c init 3 mēneši atpakaļ
cy.log 2aa98f254c init 3 mēneši atpakaļ
image.jpg 2aa98f254c init 3 mēneši atpakaļ
image.png 2aa98f254c init 3 mēneši atpakaļ
onnx2trt.py 3ffe2db4d5 new onnx models 1 mēnesi atpakaļ
result.jpg 2aa98f254c init 3 mēneši atpakaļ
test_trt.cpp 2aa98f254c init 3 mēneši atpakaļ
yolo-crowd 2aa98f254c init 3 mēneši atpakaļ
yolo-crowd-.onnx 3ffe2db4d5 new onnx models 1 mēnesi atpakaļ
yolo-crowd-n.onnx 2aa98f254c init 3 mēneši atpakaļ
yolov5-crowd-n.engine 3ffe2db4d5 new onnx models 1 mēnesi atpakaļ
yolov5.cpp 2aa98f254c init 3 mēneši atpakaļ
yolov5_trt.cpp c89936e146 加注释 3 mēneši atpakaļ

README.md

small-target-reco

解决小目标检测问题的算法集

onnx2trt.py:将onnx模型转换为tensorrt模型 onnx模型:yolo-crowd-n.onnx

yolov5_trt.cpp:使用tensorrt部署yolov5模型

训练代码仓库:https://github.com/zaki1003/YOLO-CROWD