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

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

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