草庐IT

CV-Pytorch

全部标签

PyTorch保存模型断点以及加载断点继续训练

   在训练神经网络时,用到的数据量可能很大,训练周期较长,如果半途中断了训练,下次从头训练就会很费时间,这时我们就想断点续训。一、神经网络模型的保存,基本两种方式:1.保存完整模型model, torch.save(model,save_path) 2.只保存模型的参数, torch.save(model.state_dict(),save_path) ,多卡训练的话,在保存参数时,使用model.module.state_dict()。二、保存模型训练的断点checkpoint断点dictionary中一般保存训练的网络的权重参数、优化器的状态、学习率变化scheduler的状态以及epo

如何在linux服务器上安装Anaconda与pytorch,以及pytorch卸载

如何在linux服务器上安装Anaconda与pytorch,以及pytorch卸载1,安装anaconda1.1下载anaconda安装包1.2安装anaconda1.3设计环境变量1.4安装完成验证2Anaconda安装pytorch2.1创建虚拟环境2.2查看现存环境2.3激活环境2.4选择合适的pytorch版本下载2.4.1查看cuda版本2.4.2在pytorch官网找下载命令2.5检测是否安装成功:3Pytorch卸载与重安装3.1卸载并安装指定版本参考1,安装anaconda1.1下载anaconda安装包这里我们需要在官网上查找自己需要的版本,地址链接在下面:https://

Pytorch复习笔记--nn.Conv2d()和nn.Conv3d()的计算公式

1--基本知识        nn.Conv2d()和nn.Conv3d()分别表示二维卷积和三维卷积;二维卷积常用于处理单帧图片来提取高维特征;三维卷积则常用于处理视频,从多帧图像中提取高维特征;    三维卷积可追溯于论文3DConvolutionalNeuralNetworksforHumanActionRecognition;    三维卷积使用三维卷积核,在T、H和W三个维度进行移动,以提取时间特征和空间特征,一个简单示意图如下:2--基本用法importtorchimporttorch.nnasnnif__name__=="__main__":B=8C=3T=10H=255W=25

ubuntu 18.04下安装Anaconda、Cuda、Cudnn、gpu-Pytorch

    登录服务器后,首先查看下系统版本:cat/proc/version,我们这个服务器有点特殊是arm版本的,通过nvidia-smi查看服务器显卡配置,这意味着我们要安装的cuda版本最高不能超过11.4,那么我们这里选择一个比较稳定的版本cuda11.3。1. anaconda安装:由于cpu不是intel的,需要安装arm版本的anaconda那么肯定会问,为什么要安装它呢:Anaconda是一个打包的集合,官网,它里面预装好了conda、某个版本的python、众多packages、科学计算工具等等。Anaconda支持Linux,Mac,Windows系统,提供了包管理与环境管理

大数据深度学习长短时记忆网络(LSTM):从理论到PyTorch实战演示

文章目录大数据深度学习长短时记忆网络(LSTM):从理论到PyTorch实战演示1.LSTM的背景人工神经网络的进化循环神经网络(RNN)的局限性LSTM的提出背景2.LSTM的基础理论2.1LSTM的数学原理遗忘门(ForgetGate)输入门(InputGate)记忆单元(CellState)输出门(OutputGate)2.2LSTM的结构逻辑遗忘门:决定丢弃的信息输入门:选择性更新记忆单元更新单元状态输出门:决定输出的隐藏状态门的相互作用逻辑结构的实际应用总结2.3LSTM与GRU的对比1.结构LSTMGRU2.数学表达LSTMGRU3.性能和应用小结3.LSTM在实际应用中的优势处理

复现炮哥带你学—Yolo5训练安全帽(vscode + pytorch)报错总结,数据库链接+权重文件链接

复现炮哥带你学—Yolo5训练安全帽(vscode+pytorch)报错总结,数据库链接+权重文件链接1、本篇博客说明2、训练YOLOV5前的准备工作2.1学习第四步对应的博客链接和B站链接2.2数据集下载2.3YOLOV5源代码下载2.4权重文件yolov5s.pt下载2.5放置划分好的数据集和权重文件2.6修改我们需要的数据.yaml文件2.7修改我们需要的模型.yaml文件2.8修改我们的train.py文件2.9修改我们的datasets.py文件3、开始我们的bug解决之旅3.1运行requirements.txt配置发生报错3.2运行train.py发生报错之SPPF类报错3.3运

PVN3D——WIN10 PyTorch1.8 Linemod-render最全复现

目录前言一、资源下载二、环境配置提示pytorch版本设置的原因需要安装系统cuda!!win10conda基本环境配置python-pcl配置pointnet2_ops配置Ubuntu20.04三、融合数据生成四、模型训练源码微调训练开始总结前言研究6d位姿估计,复现经典代表性论文pvn3d,因实验需求,在win10和Ubuntu20.04两个平台上实现,遇到数不清的bug,不少bug网上记录较少或记录不清楚,因此凭借记忆对bug进行记录。win10:NVIDIAGeForceRTX4090、DriverVersion:526.47Ubuntu20.04:NVIDIAGeForceRTX30

PyTorch中的matmul函数详解

PyTorch中的两个张量的乘法可以分为两种:两个张量对应的元素相乘(element-wise),在PyTorch中可以通过torch.mul函数(或者∗*∗运算符)实现两个张量矩阵相乘(Matrixproduct),在PyTorch中可以通过torch.matmul函数实现本文主要介绍两个张量的矩阵相乘。语法为:torch.matmul(input,other,out=None)函数对input和other两个张量进行矩阵相乘。为了方便后续的讲解,将input记为a,将other记为b。点积在数学中,又称数量积,是指接受在实数R上的两个1D张量并返回一个实数值0D张量的二元运算。若1D张量a

CV——基于Stitcher类实现图片拼接

[摘要]图像拼接(ImageStitching)是将同一场景的多张有重叠部分的图像拼接成一幅无缝或高分辨率图像的技术。在计算机信息技术不断发展的背景下,图像拼接技术不仅得到了有效优化,也切实应用于运动检测和跟踪、增强现实、分辨率增强、视频压缩和图像稳定等计算机视觉领域。本项目通过三种方法(SIFT算法、ORB算法、直接调用Stitcher类)实现简单的图像拼接,进行对比、分析,旨在学习计算机视觉相关领域的知识,总结图像拼接技术算法,以便后续进行深入研究,可供初学者参考。关键词:图像拼接;图像匹配;计算机视觉1.引言图像作为一种可以帮助人们直接了解世界的信息渠道,不仅可以为人们与自然界提供交互方

基于 PyTorch 的 cifar-10 图像分类

文章目录前言一、cifar-10数据集介绍二、环境配置三、实验代码1.简单网络的代码2.VGG加深网络的代码四、运行结果五、遇到的问题总结前言本文的主要内容是基于PyTorch的cifar-10图像分类,文中包括cifar-10数据集介绍、环境配置、实验代码、运行结果以及遇到的问题这几个部分,本实验采用了基本网络和VGG加深网络模型,其中VGG加深网络模型的识别准确率是要优于基本网络模型的。一、cifar-10数据集介绍cifar-10数据集由60000张分辨率为32x32彩色图像组成,共分为10类,每类包含6000张图像,cifar-10数据集有50000个训练图像和10000个测试图像。数