分享
扫一扫
超级会员免费看
自由自在的鱼丶
咕噜咕噜
展开
-
YOLOV5 TensorRT部署 BatchedNMS(engine模型推理)(下)
主要是在王新宇代码的基础上改进，引入对BatchedNMS的解码。原创 2024-04-25 15:50:23 · 427 阅读 · 0 评论 -
YOLOV5 TensorRT部署 BatchedNMS(转换engine模型)(上)
yolov5 tensorrt部署时引入nms plugin，nms在cuda上执行，减少内存拷贝和运行时间原创 2024-04-23 15:11:55 · 523 阅读 · 1 评论 -
yolov5 obb旋转框 tensorrt部署
yolov5-obb tensorRT部署代码结合王新宇和fish-kong两者的代码，可以多batch批量检测旋转框。等待生成engine文件。执行完毕生成推理结果。原创 2023-11-06 18:40:40 · 727 阅读 · 0 评论 -
Yolov5旋转框(斜框)检测自己的数据集，附带代码+数据集(模型可以收敛)
yolov5检测旋转目标，输出旋转框原创 2023-11-01 10:27:58 · 2689 阅读 · 6 评论 -
yolov5使用最新MPDIOU损失函数，有效和准确的边界盒回归的损失，优于GIoU/EIoU/CIoU/EIoU（附代码可用）
MPDIOU 能够有效解决重叠和包含问题，并且能够有效地处理预测框和目标框长宽比相同的情况，而且计算量小，模型训练速度快。在数据集上测试和验证，发现能够提升训练精度，并且有效地区分两个目标存在包含的关系的预测框回归，推荐大家使用MPDIOU 进行训练。原创 2023-09-18 19:01:20 · 2696 阅读 · 0 评论 -
yolov5+Repulsion损失函数，解决密集遮挡问题(附带代码可用)
yolov5+RepLoss，解决密集遮挡问题原创 2023-09-12 19:47:13 · 3627 阅读 · 24 评论