yolov5-6.0网络添加小目标检测头 TensorRT部署

小目标检测使用TPH-YOLOv5 中的yolov5l-xs-tph.yaml模型；对其中的C3STR替换为C3模块；本文的TensorRT部署是基于yolov5l-xs-tph修改后的模型训练部署，其精度一般主要学习如何TensorRT部署，是在wang-xinyu基础上简单修改；

TPH-YOLOv5 ：https://github.com/cv516Buaa/tph-yolov5
yolov5l-xs-tph.yaml模型修改后：tensorrt-nyy: ​yolov5-6.0网络添加小目标检测头 TensorRT部署；​

1、模型训练
具体修改在head 部分 anchors改为4； C3STR改为C3

# YOLOv5 声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop】
推荐阅读
article从PyTorch导出ONNX使用TensorRT模型加速_从 pytorch 转换为 tensorr...
LInux下TensorRT加速流程_从 pytorch 转换为 tensorrt 并加速推理从 pytorch 转换为...
                
                    赞
踩
articleTensorRT 下不同 API 推理时间的对比实验_tensorrt api...
主要对比TensoRT C++ api、python api，在onnxparser和原生api构建网络方面的加速效果_...
                
                    赞
踩
articleonnx和pytorch，tensorrt 推理速度对比GPU CPU_tensorrt cpu...
通常我们在训练模型时可以使用很多不同的框架，比如有的同学喜欢用Pytorch，有的同学喜欢使用TensorFLow，也有...
                
                    赞
踩
articlepytorch gpu推理、onnxruntime gpu推理、tensorrt gpu推理比较，及...
onnx , tensorrt 安装推理比较，有代码_onnxruntime gpuonnxruntime gpu   ...
                
                    赞
踩
articlePyTorch 2.0 推理速度测试：与 TensorRT 、ONNX Runtime 进行对比_p...
我们使用 Nebuly 的开源库 Speedster 运行了一些推理测试，对于这个我们这个测试，Speedster 允许...
                
                    赞
踩
articleonnx转tensorrt中的问题记录_this version of tensorrt only ...
onnx转tensorrt_this version of tensorrt only supports int32!t...
                
                    赞
踩
articleOnnxRuntime TensorRT OpenCV::DNN性能对比(YoloV8)实测_yol...
cpu选择onnxruntime或者dnn都可以,建议选择ort.gpu选择tensorrt,如果有兼容需求就只能选择o...
                
                    赞
踩
articleyolov5-pytorch模型转onnx模型再转tensorflow savedmode模型进行t...
pytorch框架下的yolov5模型部署到tensorflow serving，需要将pytorch的pt模型转换为o...
                
                    赞
踩
articleyolov5 simplepose人体姿态...
先给效果图：有时快点，100ms，有时慢，200ms，精度不是特别准。github地址：https://github.c...
                
                    赞
踩
article涨点技巧：Yolov5/Yolov7引入CVPR2023 Demystify Transformer...
特征转换模块（例如注意力或卷积）之间的主要区别在于空间特征聚合的方式，即所谓的“空间标记混合器”（STM）。因此制定了一...
                
                    赞
踩
article在 C++ 和 Python 中使用 YOLOv5 OpenCV DNN 进行对象检测_c++ yo...
在 C++ 和 Python 中使用 YOLOv5 OpenCV DNN 进行对象检测实现及相关代码介绍_c++ yol...
                
                    赞
踩
articleYOLO系列详解：YOLOv1、YOLOv2、YOLOv3、YOLOv4、YOLOv5、YOLOv6...
一、前言YOLO系列是one-stage且是基于深度学习的回归方法，而R-CNN、Fast-RCNN、Faster-RC...
                
                    赞
踩
article【深度学习】yolov5+deepsort 完成计数和行人行人重识别的追踪_车流计数yolo...
行人重识别是计算机视觉的基本任务之一，首先要有一个detector（检测器来检测到目标），然后将检测到的目标送入到tra...
                
                    赞
踩
articleAI全栈代码下载 | YOLO系列/车道线/3D目标检测+改进/量化/蒸馏/剪枝/TensorRT...
大家好，我是Leo，今天和大家介绍和推荐一下【集智书童】知识星球学习社区，这里不仅仅适用于初学者的基础学习，同时也适合已...
                
                    赞
踩
articleyolov5 VS yoloX VS yolov8_yolox和yolov5区别...
实验笔记_yolox和yolov5区别yolox和yolov5区别              YOLOv5-6.0版本的...
                
                    赞
踩
article三万字硬核详解：yolov1、yolov2、yolov3、yolov4、yolov5、yolov7_...
Yolo (You Only Look Once) 是目标检测 one-state 的神经网络模型，可以在图像中找出特定...
                
                    赞
踩
articleyoloV5 增加小目标检测层_yolov5增加检测头...
原始网络有三个检测头： 分别是 80x80 （小目标） 40x40（中目标） 160x160(小目标)，现在要增加小目标...
                
                    赞
踩
相关标签
深度学习
人工智能
计算机视觉
pytorch
python
神经网络
tensorrt
onnx
c++
yolov5 pose
YOLO
opencv
dnn