深度学习模型C++部署TensorRT
如何最简单的在无CUDA环境的机器上利用Tensorrt部署深度学习模型。
【魔改YOLOv5-6.x(4)】结合EIoU、Alpha-IoU损失函数
文章目录前言EIoU论文简介加入YOLOv5Alpha-IoU论文简介加入YOLOv5References前言本文使用的YOLOv5版本为v6.1,对YOLOv5-6.x网络结构还不熟悉的同学,可以移步至:【YOLOv5-6.x】网络模型&源码解析想要尝试改进YOLOv5-6.1的同学,可以
yolov7配置与训练记录(二)
yolov7配置与训练记录(一) 已经完成了环境的配置,下面开始文件内部的操作yolov7官方下载地址为1 将下载好的预训练权重放在内需要在yolov7中新建weights文件夹(也是为了方便管理权重文件)如果未报错,则说明成功需要在yolov7中新建datasets文件夹(也是为了方便管理训练数据
Multihead Attention - 多头注意力
多头注意力的基本概念及实现
深度学习训练营之yolov5训练自己的数据集
这个yolov5的训练总算是弄出来了,期间遇到了很多的报错,不过也算是学到了许多
浅析Swin transformer模型(通俗易懂版)
对于最近新出的Swin Transformer的系统学习,包括模型的基本结构、参数介绍、计算过程等详细介绍,全面了解该模型,文中包含相关代码和论文下载连接。
基于RGB-D图像的多模态特征融合
几种融合介绍
计算机领域各大顶会顶刊集合梳理
每一个领域内,都有很多优秀的、认可度高的会议或者期刊。其他领域不太了解,但一般用SCI一区,二区等来区分论文质量。对于计算机领域而言,一般的分类方式是CCF评级,从A到C含金量依次降低。为了更好地关注计算机各个相关领域的最新技术、学术动态,今天梳理一下计算机领域内的顶会顶刊。会议会议论文指的是以被会
Colossal-AI的安装
是一个集成的大规模深度学习系统,具有高效的并行化技术。最近在学习stable diffusion model,但是这个模型成本比较高,作为低端学习者,借助colossal-ai加速训练,即能满足显卡要求又能节约时间。下载并解压缩,进入ColossalAI/ -> examples/ -> tutor
搭建自己的语义分割平台deeplabV3+
搭建deeplabV3+网络 数据准备 源码修改 训练 测试
21世纪20年代的ConvNet——ConvNeXt
基于论文《21世纪20年代的ConvNeXt》的翻译以及一些思考总结
基于BiLSTM-Attention实现天气变量预测风速
风速预报是预警灾害性天气的一项重要任务,本项目使用循环神经网络BiLSTM-Attention训练一个网络模型,来预测在给定指定日期的天气变量来预测对应的风速情况。
【深度学习】目标检测的性能评价指标,mAP_0.5,mAP_0.5,0.95,0.05
计算mAP之前先考虑我们有的数值:图片原label的bbox、模型预测的bbox、模型预测的bbox的置信度、模型预测的bbox中目标类别的分类置信度。此外,我们还需要确定“IoU数值阈值”和“置信度阈值”,模型的预测能满足“IoU数值阈值”与“置信度阈值”(NMS算法)的结果参与最终混淆矩阵计算。
3DResNet 学习记录
近期同时在进行的两个深度学习项目都需要用到3DResNet模型,本着不做调包侠的心态,还是要好好把模型的原理看一看的。1、ResNet结构理解首先先理解一下二维的ResNet吧。
Pytorch加载模型只导入部分层权重,即跳过指定网络层的方法
需求Pytorch加载模型时,只导入部分层权重,跳过部分指定网络层。(权重文件存储为dict形式)本文参考总结他人的做法,主要是load时加if判断哪些网络层不需要,或者直接strict=False,跳过没有的网络层。还有对载入的参数更新有具体要求的方法,固定参数、或者不同参数有不同的更新速度。同时
【论文笔记】CycleGAN(基于PyTorch框架)
CycleGAN(基于PyTorch框架)
YOLOv5实战之输电线路绝缘子缺陷检测识别
本教程主要是利用YOLOv5算法实现对输电线路绝缘子缺陷进行检测识别。通过无人机搭载相机头云台对输电线路上的绝缘子进行数据采集,挑选出绝缘子上有故障的图片数据,共2000张左右图片,输电线路绝缘子缺陷数据集中的部分图片如下图所示。对收集到的2000张左右绝缘子缺陷数据集进行数据标注, 标注了3种
Openpcdet训练自己的数据集
本人移植其他的数据集,由于我有自己的image数据,已经按照kitti数据集的格式转换为velodyne, calib, label, image四个文件,并且实现了评估,以及最终的检测结果,所以可能和其他博主不一样。这个函数主要用来生成你的数据字典,一般以.pkl后缀,如果你不需要评估,可以将其中
机器学习笔记 - 什么是图注意力网络?
顾名思义,图注意力网络是图神经网络和注意力层的组合。要理解图注意力网络,我们首先需要了解什么是注意力层和图神经网络。首先,我们将看一下对图神经网络和注意力层的基本理解,然后我们将重点介绍两者的结合。让我们看一下图神经网络。图神经处理是数据科学和机器学习领域研究的热点之一,因为它们具有通过图数据进行学
深度学习之快速实现数据集增强的方法
我们在深度学习训练之前准备数据集的时候,特别是打标注的数据集,需要耗费大量的人力物力打标签,在打完的基础我们还可以直接对数据集进行二次增强,即**数据集增强**。