使用YOLOv8训练滑坡大规模多传感器滑坡检测数据集
n/ ├── val/ └── test/ 每个文件夹中包含对应的图像文件和掩码文件。确保所有图像文件都是.jpg格式，而掩码文件是.png格式，并且它们的名字与对应的图像文件相同。 数据集转换 由于YOLOv8需要标签文件为YOLO格式（即.txt文件），我们需要将Mask格式的标签文件转换为YOLO格式。 安装依赖 确保你已经安装了必要的库： bash 深色版本 pip install numpy op...
使用onnxruntime c++ API实现yolov5m视频检测
@[使用onnxruntime c++ API实现yolov5m视频检测] 本文演示了yolov5m从模型导出到onnxruntime推理的过程 一.创建容器 docker run --shm-size=32g -ti --privileged --net=host \ --rm \ -v $PWD:/home -w /home ghcr.io/intel/llvm/ubuntu2204_base /bi...
如何使用yolov8训练——电力设备漏油检测图像数据集，数据集总数为687张，标注为voc格式，也可以转yolo格式
电力设备漏油检测图像数据集，数据集总数为687张，标注为voc格式，也可以转yolo格式 如何使用YOLOv8训练电力设备漏油检测图像数据集的详细步骤和代码。我们将使用PyTorch和YOLOv8的官方实现来进行训练。 项目结构 OilLeakDetection/ ├── data/ │ ├── images/ │ │ ├── train/ │ │ ├── val/ │ │ └── test/ │ ├──...
YoloV9改进策略：BackBone改进|CAFormer在YoloV9中的创新应用，显著提升目标检测性能
摘要 在目标检测领域，模型性能的提升一直是研究者和开发者们关注的重点。近期，我们尝试将CAFormer模块引入YoloV9模型中，以替换其原有的主干网络，这一创新性的改进带来了显著的性能提升。 CAFormer，作为MetaFormer框架下的一个变体，结合了深度可分离卷积和普通自注意力机制的优势。在底层阶段，CAFormer采用深度可分离卷积作为令牌混合器，有效降低了计算复杂度并保持了良好的性能；而在顶...
YOLOv5复现（论文复现）
YOLOv5复现（论文复现） 文章目录 YOLOv5复现（论文复现）概述模型结构正负样本匹配策略损失计算数据增强使用方式训练测试验证Demo 概述 模型结构 # CSPDarkNetclass CSPDarkNet(nn.Module): def __init__(self, depth=1.0, width=1.0, act_type='silu', norm_type='BN', depthwise...
YoloV8改进策略：BackBone改进|CAFormer在YoloV8中的创新应用，显著提升目标检测性能
摘要 在目标检测领域，模型性能的提升一直是研究者和开发者们关注的重点。近期，我们尝试将CAFormer模块引入YoloV8模型中，以替换其原有的主干网络，这一创新性的改进带来了显著的性能提升。 CAFormer，作为MetaFormer框架下的一个变体，结合了深度可分离卷积和普通自注意力机制的优势。在底层阶段，CAFormer采用深度可分离卷积作为令牌混合器，有效降低了计算复杂度并保持了良好的性能；而在顶...
YOLOv5改进：Unified-loU，用于高品质目标检测的统一loU ，2024年8月最新IoU
💡💡💡现有IoU问题点：IoU (Intersection over Union)作为模型训练的关键，极大地显示了当前预测框与Ground Truth框之间的差异。后续研究者不断在IoU中加入更多的考虑因素，如中心距离、纵横比等。然而，仅仅提炼几何差异是有上限的;而且新的对价指数与借据本身存在潜在的联系，两者之间的直接加减可能会导致“对价过高”的问题 💡💡💡本文独家改进：提出了一种新的IoU损失函数，称为...
YOLOv1代码复现（论文复现）
YOLOv1代码复现（论文复现） 文章目录 YOLOv1代码复现（论文复现）论文介绍主要内容实验部分卷积网络结构计算损失核心代码 缺点 论文介绍 主要内容 实验部分 卷积网络结构 计算损失 核心代码 class ResNet(nn.Module): def __init__(self, block, layers): super(ResNet, self).__init__() # 通道数64 self....
yolov8实例分割重要图片
训练分割要准备好数据集和分割预训练权重文件 下面这张图是数据集的格式 下面这张图配置数据集，下面names 要和labelme转txt里配置的一样 下面这张图进行训练，配置一些全局参数 ，初始的yolov8s-seg.pt文件需要到github上yolov8开源项目里下 labelme转txt # -*- coding: utf-8 -*- import osimport numpy as npim...
YoloV10改进策略：BackBone改进|RIFormer在YoloV10中的创新应用与显著性能提升
摘要 在深度学习领域，模型架构的不断优化是推动计算机视觉任务性能飞跃的关键驱动力。近期，我们创新性地将高效的RIFormer主干网络引入到了YoloV10目标检测模型中，这一变革不仅保留了YoloV10原有的高速推理能力，更在检测精度上实现了显著提升，为实时目标检测任务树立了新的标杆。 RIFormer主干网络简介： RIFormer是一种经过精心设计的视觉骨干网络，其核心在于去除了传统视觉Transfo...