深度学习
文章平均质量分 63
论搬砖的艺术
每天进步一点点。
展开
-
Yolov8-车辆跟踪(BoT-SORT和ByteTrack算法)
ByteTrack预测的速度感觉比BoT-SORT快一些,流畅一些。原创 2024-03-11 14:39:27 · 694 阅读 · 1 评论 -
Yolov8将.pt文件转换为tensorRt的.trt文件(模型部署)
【代码】Yolov8将.pt文件转换为tensorRt的.trt文件(模型部署)原创 2024-03-10 16:09:10 · 791 阅读 · 0 评论 -
基于yolov8与pyqt5的火焰烟雾实时检测系统设计
界面权重:可以选择自己训练的yolov8模型,也可以用一些改进的yolov8模型作为系统的权重。功能:单张图片的检测,视频文件的检测,多张图片同时检测,以及摄像头实时检测。调整:可以调整置信度,IOU以适应更好的输出。保存结果:可以选择是否保存预测结果到本地。GPU预测:可以将预测调到gpu上进行预测,这样实时检测更加的顺畅。在cpu上预测,延时将近110ms左右,对比在gpu上进行预测,延时只有14ms。图为摄影头实时检测效果图。原创 2024-03-02 22:57:22 · 585 阅读 · 0 评论 -
yolov8-更换卷积模块-ContextGuidedBlock_Down
原创 2024-02-28 11:15:12 · 640 阅读 · 0 评论 -
yolov8添加注意力机制模块-ShuffleAttention
函数用于对张量的维度进行重新排列。这里,它将分组的维度(索引为1的维度)和通道的维度(索引为2的维度)调换位置。经过这一步操作后,张量的形状将变为。,其中分组内的通道现在已经被混洗。由于维度调换的操作,原先属于同一组的通道现在分散到了不同的位置,从而完成了通道混洗的过程。被重新塑形(reshape)为一个新的形状。表示自动计算该维度的大小,具体来说,这里的。:在上一步维度调换之后,此行代码再次将张量。进行重新塑形,使其变回原始的4维形状。:最后,返回了经过通道混洗后的张量。组,每组具有相等数量的通道。原创 2024-02-26 21:34:08 · 960 阅读 · 0 评论 -
yolov8添加注意力机制模块-CBAM
3.修改yolov8.yaml文件位置(ultralytics-main/ultralytics-main - attention/ultralytics/cfg/models/v8/yolov8.yaml)。修改head模块,修改的内容如下图。4.测试打印网络。已经添加成功。原创 2024-02-25 00:08:24 · 2085 阅读 · 1 评论 -
yolov8源码解读Detect层
关于网络的backbone,head,以及detect层后处理,可以参考文章结尾博主的文章。原创 2024-02-17 00:20:39 · 974 阅读 · 2 评论 -
铁轨语义分割(Unet结合resnet系列)
一类是图片,一类是图像标签。原创 2024-02-01 15:48:54 · 702 阅读 · 0 评论 -
yolov8上使用gpu教程
在选择组件的时候,将CUDA中的Nsight VSE和Visual Studio Integration取消勾选,后选择下一步,即可安装成功。去pytorch官网,下载对应的版本。最好下载cu+的,pip命令的。因为下其他的我不知道为什么失败了。找到cuda安装目录,将这三个文件夹下的文件全部移到cuda对应的文件夹下。安装的前提基础是,有vs的C++环境。4.取消勾选,这里就借用了其他博主的一些图。1.查看支持的cuda版本,并去官网下载。找到安装目录,将这两个添加进去。5.测试gpu,为True即可。原创 2024-01-25 23:01:41 · 4527 阅读 · 0 评论 -
yolov8进行预测代码
【代码】yolov8进行预测代码。原创 2024-01-12 19:43:01 · 611 阅读 · 0 评论 -
yolov8训练自己的数据集
进入项目目录,打开命令台,pip安装ultralytics库。依赖环境只需要安装ultralytics库即可,yolov8已经将所需要的依赖集成到了ultralytics库。分类数根据需要调整yolov8.yaml文件。需要用yolov8的n,s,m,l,x。即从yaml创建模型时添加对应的字母即可。注意,这里可以自己写一个简单的py文件进行预测。python版本需要>=3.7才可以。并将预训练权重放在项目文件夹下。训练函数自己写一个简单的。原创 2024-01-03 11:35:43 · 828 阅读 · 0 评论 -
Intel Image Classification With Pytorch
Intel Image Classification With Pytorch原创 2023-02-08 00:12:04 · 356 阅读 · 1 评论 -
pytorch softmax回归从零开始实现
pytorch softmax回归从零开始实现原创 2023-01-01 22:03:00 · 154 阅读 · 0 评论 -
线性回归的简洁实现
线性回归的简洁实现原创 2022-12-14 16:16:53 · 132 阅读 · 0 评论 -
梯度下降算法
梯度下降算法原创 2022-10-26 16:42:15 · 200 阅读 · 0 评论 -
Django 分页组件的封装(靓号管理)
Django 分页组件的封装(靓号管理)原创 2022-09-28 22:32:25 · 313 阅读 · 0 评论 -
python 获取目录下的文件并按比例划分
python 获取目录下的文件并按比例划分原创 2022-09-26 23:03:39 · 556 阅读 · 0 评论 -
新闻文本分类
新闻文本分类原创 2022-07-28 21:17:08 · 372 阅读 · 0 评论 -
中文语义匹配
中文语义匹配原创 2022-07-28 16:02:49 · 568 阅读 · 2 评论 -
基于TNEWS‘ 今日头条中文新闻(短文本)分类
基于TNEWS' 今日头条中文新闻(短文本)分类原创 2022-07-27 12:06:01 · 1844 阅读 · 4 评论 -
resnet152 辣椒病虫害图像识别1.0
resnet152 辣椒病虫害图像识别1.0原创 2022-06-26 20:26:12 · 1583 阅读 · 6 评论 -
plt.gca()画框及打标签
plt.gca()画框及打标签原创 2022-06-24 14:56:15 · 3162 阅读 · 0 评论 -
VGG 中草药识别
VGG 中草药识别原创 2022-06-22 20:09:58 · 1653 阅读 · 1 评论 -
基于Attention U-Net的宠物图像分割
基于Attention U-Net的宠物图像分割原创 2022-06-18 16:10:10 · 1041 阅读 · 0 评论 -
基于paddlepaddle的新冠肺炎识别
基于paddlepaddle的新冠肺炎识别原创 2022-06-12 20:39:25 · 411 阅读 · 0 评论 -
基于tensorflow的校园绿植识别
基于tensorflow的校园绿植识别原创 2022-06-11 21:37:30 · 455 阅读 · 2 评论 -
基于tensorflow的猫狗分类
基于tensorflow的猫狗分类原创 2022-06-06 16:46:26 · 1364 阅读 · 2 评论 -
基于tensorflow的手写数字识别
基于tensorflow的手写数字识别原创 2022-06-02 20:55:12 · 2089 阅读 · 2 评论 -
卷积神经网络初识
卷积神经网络初识原创 2022-04-27 21:52:40 · 1667 阅读 · 0 评论 -
反向传播算法初识
反向传播算法初识原创 2022-04-26 16:56:59 · 214 阅读 · 0 评论 -
前馈神经网络
前馈神经网络原创 2022-04-20 18:45:40 · 3678 阅读 · 0 评论