保研笔记八——YOLOV5项目复习

学习转载自:睿智的目标检测56——Pytorch搭建YoloV5目标检测平台_Bubbliiiing的博客-CSDN博客_睿智yolo Pytorch 搭建自己的YoloV5目标检测平台(Bubbliiiing 源码详解 训练 预测)-主干网络介绍_哔哩哔哩_bilibili还有一些视频的学习笔记。

关于Pytorch中的train()和eval()(以及no_grad())

这三个函数实际上很常见,先来简单看下使用方法train()是nn.Module的方法,也就是你定义了一个网络model,那么表示将该model设置为训练模式,一般在开始新epoch训练时,我们会首先执行该命令:同train()一样,其用法和含义也一样,eval()是nn.Module的方法,也就是你

LSTM实现多变量输入多步预测(Seq2Seq多步预测)时间序列预测(PyTorch版)

本专栏整理了《深度学习时间序列预测案例》,内包含了各种不同的基于深度学习模型的时间序列预测方法,例如LSTM、GRU、CNN(一维卷积、二维卷积)、LSTM-CNN、BiLSTM、Self-Attention、LSTM-Attention、Transformer等经典模型,包含项目原理以及源码,每一

知识蒸馏算法和代码(Pytorch)笔记分享,一个必须要了解的算法

知识蒸馏算法和代码(Pytorch)笔记分享,一个必须要了解的算法

MMdetection之train.py源码详解

目录一、tools/train.py二、源码详解三、核心函数详解(一)build_detector(mmdet/models/builder.py)(二)build_dataset(mmdet/datasets/builder)(三)train_detector(mmdet/apis/train.p

Yolov5--从模块解析到网络结构修改(添加注意力机制)

文章目录1.模块解析(common.py)01. Focus模块02. CONV模块03.Bottleneck模块:04.C3模块05.SPP模块2.为yolov5添加CBAM注意力机制最近在进行yolov5的二次开发,软件开发完毕后才想着对框架进行一些整理和进一步学习,以下将记录一些我的学习记录。

pytorch从零开始搭建神经网络

pytorch教程之nn.Sequential类详解——使用Sequential类来自定义顺序连接模型_LoveMIss-Y的博客-CSDN博客_sequential类pytorch教程之nn.Module类详解——使用Module类来自定义模型_LoveMIss-Y的博客-CSDN博客_torch

YOLOv7中的数据集处理【代码分析】

本文章主要是针对yolov7中数据集处理部分代码进行解析(和yolov5是一样的),也是可以更好的理解训练中送入的数据集到底是什么样子的。数据集的处理离不开两个类,(from torch.utils.data.dataloader import DataLoader),不论什么样的算法,在处理数据集

【Anaconda创建虚拟环境】报错及解决办法

Anaconda创建虚拟环境的一些报错问题及解决办法记录

PyTorch深度学习实战 | 基于ResNet的人脸关键点检测

人脸关键点检测指的是用于标定人脸五官和轮廓位置的一系列特征点的检测,是对于人脸形状的稀疏表示。关键点的精确定位可以为后续应用提供十分丰富的信息。因此,人脸关键点检测是人脸分析领域的基础技术之一。许多应用场景(如人脸识别、人脸三维重塑、表情分析等)均将人脸关键点检测作为其前序步骤来实现。本文将通过深度

Easy Deep Learning——卷积层

由于卷积核尺寸可以远远小于输入尺寸,即减少需要学习的参数的数量,并且针对每个卷积层可以使用多个卷积核获取输入的特征映射,对数据(尤其是图像)具有很强的特征提取和表示能力,并且在卷积运算之后,使得卷积神经网络结构对输入的图像具有平移不变的性质。下面使用一张图像来展示经过卷积后,输出的特征映射的结果。在

深度学习之图像分类(十八)-- Vision Transformer(ViT)网络详解

深度学习之图像分类(十八)Vision Transformer(ViT)网络详解目录深度学习之图像分类(十八)Vision Transformer(ViT)网络详解1. 前言2. ViT 模型架构2.1 Embedding 层2.2 Transformer Encoder 层2.3 MLP Head

一天学会应用GAN扩充数据集(pytorch)

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档文章目录前言一、GAN是什么?二、实现1.总结前言提示:这里可以添加本文要记录的大概内容:例如:随着人工智能的不断发展,机器学习这门技术也越来越重要,很多人都开启了学习机器学习,本文就介绍了机器学习的基础内容。提示:以下是本篇文章正文内

利用pytorch 模型载入部分权重

本文介绍如何在pytorch中载入模型的部分权重第1个常见的问题: 在分类网络中,当载入的预训练权重的全连接层与我们自己实例化模型的节点个数不一样时,该如何载入?比如在花卉数据集分类时只有5类,所以最后一层全连接层节点个数为5,但是我们载入的预训练权重是针对ImageNet-1k的权重,它的全连接层

基于pytorch使用LSTM进行文本情感分析

下载链接项目中使用的模型是LSTM,在模型中我们定义了三个组件,分别是embedding层,lstm层和全连接层。Embedding层:将每个词生成对应的嵌入向量,就是利用一个连续型向量来表示每个词Lstm层:提取语句中的语义信息Linear层:将结果映射成2大小用于二分类,即正反面的概率注意:在L

Pytorch中torch.sort()和torch.argsort()函数解析

torch.sort(),如下图所示:输入input,在dim维进行排序,默认是dim=-1对最后一维进行排序,descending表示是否按降序排,默认为False,输出排序后的值以及对应值在原输入imput中的下标3.1 dim = -1 表示对每行中的元素进行升序排序,descending=F

DBNet实战:详解DBNet训练与测试(pytorch)

论文连接:https://arxiv.org/pdf/1911.08947.pdfgithub链接:github.com网络结构首先,图像输入特征提取主干,提取特征;其次,特征金字塔上采样到相同的尺寸,并进行特征级联得到特征F;然后,特征F用于预测概率图(probability map P)和阈值图

pytorch对网络层的增,删, 改, 修改预训练模型结构

1.我们使用vgg11网络做示例, 看一下网络结构:加载本地的模型:vgg16 = models.vgg16(pretrained=False) #打印出预训练模型的参数vgg16.load_state_dict(torch.load('vgg16-397923af.pth'))加载库中的模型imp

[九]深度学习Pytorch-transforms图像增强(剪裁、翻转、旋转)

深度学习Pytorch-transforms图像增强(剪裁、翻转、旋转)