Dawn向阳而生-CSDN博客

由于我最近在学习libtorch相关的东西，所以就记录一下使用libtorch与pytorch一样对张量操作的语法。下面是我转载的一位大佬的文章，这里只做技术探讨，不做其他用途。想要学习的可以参考一下大佬的文章，我使用的时候也是对文章里面少量的语法进行了验证，读者可以自己尝试的，libtorch对张量(tensor)的切片、索引操作下面还有一个libtorch常用的函数，其实和pytorch是对照的，也可以参考一下，功能是都是一样的，只是实现的方法不一样，很少有人使用libtorch去训练，太伤人了

2022-05-13 10:18:42 549

原创 yolov5-pytorch导出模型问题

yolov5导出模型目的及参数设置

2022-05-07 14:34:02 899

原创 QT学习之路-记事本

1、在创建记事本之前先明白有哪些功能：先创建一个菜单栏，菜单栏是用来装各种功能的一个地方如上图所示，文件、编辑所在的地方为菜单栏，//创建菜单栏 QMenuBar *menubar=menuBar();菜单栏创建好后再创建菜单，菜单在菜单栏里面，那么很明显，“文件”，“编辑”等就是菜单。创建命令如下：//创建菜单 QMenu *file=new QMenu("文件",this); menubar->addMenu(file); QMenu*editor=n

2022-04-10 21:40:50 1784 1

原创 libtorch-加载预训练模型出现No such serialized submodule: ‘xxx‘

今天在用libtorch训练得时候，想用预训练模型加速训练，居然报错，以为是模型得问题，然后重新训练了一个模型，作为预训练模型，还是报错，一时找不到原因，最后在大佬的帮助下搞定了，先贴上错误的：这是我以为的，实际上应该这样：这里有一些torch::load和from_load应该有很大差别，具体什么差别呢，等有时间在查，只知道前者用在部署的时候...

2022-04-01 17:11:56 241

转载【无标题】 libtorch C++ vs2017 debug模式可以正确加载模型，release模式错误

转发：libtorch C++ vs2017 debug模式可以正确加载模型，release模式错误https://blog.csdn.net/weixin_43862688/article/details/120533390

2022-03-29 09:33:41 200

原创使用vs2019将libtorch或者网络打包成dll

最近在搞一个程序，需要把libtorch打包成dll，让别人使用，在网上找了好久都没只有找到合适的答案，最后在一个博主大哥那里找到了，很是感激这位大哥与我们分享自己的经验，接下来我把我自己的学习过程也与大家分享一下。一、打开vs2019，创建新项目，选择创建动态链接库。2、配置新项目使用Debug和x64，为什么是debug，因为我的libtorch是DeBug版本的，为什么是x64，因为x86性能支持不了。3、不使用预编译头4、新建头文件和cpp文件在‘LibtorchTest.h’

2022-03-20 16:54:47 1327 2

原创 c++ 关于error LINK2005: XXXX已经在 .obj 中定义“的问题

代码很多的时候出现这个问题很棘手，小号大量时间来排查问题，所以一边学习一边记录学习中出现的问题。原因：是因为编程习惯的问题，直接在头文件.h中直接定义了变量或者函数，然后A.cpp包含了.h,B.cpp也包含了.h生成A.obj，B.obj，但是 .h中定义的变量会在这两个obj中又被定义，所以，导致了这个LINK2005错误。没有包含也会出现这样的错误，尽量不要在头文件中定义。...

2022-03-10 18:23:27 3332 3

原创 Yolov5 -libtorch部署

将python训练好的网络模型转为可以再libtorch上走的pth模型，转换代码如下：可以直接赋值粘贴import argparseimport sysimport timesys.path.append('./') # to run '$ python *.py' files in subdirectoriesimport torchimport torch.nn as nnimport modelsfrom models.experimental import attempt

2022-03-01 18:16:25 681

原创 libtorch-resnet18

与大家分享一下自己在学习使用libtorch搭建神经网络时学到的一些心得和例子，记录下来供大家参考首先我们要参考着pytorch版的resnet来搭建，这样我们可以省去不必要的麻烦，上代码：1、首先是pytorch版残差模块class ResidualBlock(nn.Module): def __init__(self, inchannel, outchannel, stride=1, shortcut=None): super(ResidualBlock, self).__

2022-01-05 15:16:52 1947

原创 Dropout 丢弃函数的使用

Class USeDropout(nn.Module): def __init__(self): super(DropoutFC, self).__init__() self.fc = nn.Linear(100,20) self.dropout = nn.Dropout(p=0.5)#声明 def forward(self, input): out = self.fc(input)

2021-12-21 18:24:27 394

原创 vgg16网络改深度可分离卷积

原网络class CNN0(nn.Module): def __init__(self,imageHeight,nChannel): super(CNN0,self).__init__() assert imageHeight % 32 == 0,'image Height has to be a multiple of 32' self.conv0 = nn.Conv2d(in_channels=nChannel,out_channels=

2021-12-21 17:33:45 500

转载转载知乎-深度可分离卷积

深度可分离卷积

2021-12-21 16:55:01 169

转载卷积神经网络的入门与提升

转载一篇文章，感觉很好，大家有时间可以读一下，每次读都有新体会卷积神经网络的入门

2021-12-20 15:00:38 85

原创 python学习 if is none else

本人功力很浅，所以有大佬的代码中是用简洁大方式写的，我需要学习一下residual = x if self.right is None else self.right(x)上边的这行代码等价于下面的：if self.right: residual=x else: residual=self.right(x)相比于上面的，我更喜欢用下面的方式，虽然没有上面的简洁，但是这样可以让像我一样的人更直观的了解。学到就是赚到。...

2021-12-20 11:57:12 2888 5

原创 pytorch-resnet34残差网络理解

工欲善其事必先利其器，在使用该网络之前要先了解该网络的具体细节，我今天也是第一次查资料，然后加上自己的理解去写这篇学习成长文章。残差模块class ResidualBlock(nn.Module): def __init__(self, inchannel, outchannel, stride=1, dowansample=None): super(ResidualBlock, self).__init__() self.left = nn.Sequential

2021-12-20 11:40:27 3325

原创 pytorch版crnn网络框架

这是我们常见的pytorch版的crnn网络框架，我遵循老师的想法，用C++和libtorch也搭建了一个一模一样的框架，主要是用来部署用的，我暂时把libtorch版本的代码放到了私密文章里面，还没有公开，有需要的私信我import torch.nn as nnimport torch.nn.functional as Ffrom collections import OrderedDictclass BidirectionalLSTM(nn.Module): # Inputs hidde

2021-12-15 10:24:02 1853

原创 python从txt中批量读取图片名后将图片另存为

我们在做深度学习的时候几乎都需要用到txt文件，还有对应的图片名，这个时候如果数据就几十个还好，如果再增多的话，我们就受不了了，下面我就记录顺便保存一下代码txt_file="C:\\Users\\ljs\\Desktop\\rename\\test_501.txt" #这里是txt文件的地址a=[] #创建一个列表with open(txt_file, 'r', encoding='utf-8') as file: for c in file.readlines(): a.a

2021-11-06 12:06:13 870

原创数据集不够-数据增强-添加高斯噪声

import cv2import osimport numpy as npimport random#添加高斯噪声def gaussian_noise(img,mean,sigma): ''' 此函数将产生高斯噪声加到图片上 :param img:原图 :param mean:均值 :param sigma:标准差 :return:噪声处理后的图片 ''' img = img/255 #图片灰度标准化 noise = n

2021-11-02 14:09:39 5450 2

原创 CRNN+CTCocr识别采坑记录

最近在做ocr识别项目，由于是第一次接触这个东西，所以在整个过程中采到了非常多的坑，这里就记录一下采坑的地方，以及都是怎么解决的，方便大家接下来使用。1、首先是这个坑pytorch错误：RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED （安装cuda）这个错误表达的意思好像是内存或者显存不够，这个一般是在训练开始几轮之后才会发生，这个时候就要去检查train时传入hatch_size和test的batch_size，一般这个这两个是同时

2021-11-01 09:17:32 560 2

原创 yolov5（pytorch）实现目标检测后输出检测框的位置信息

在yolov5检测目标后想要输出目标位置，在网上找了几个简称，都没有实现，第一个是让找到plot_one_box的这个函数，然后去修改，但是发现没有鸟用。最后在以为博友哪里找到了正确的解决方法，就是在detect.py文件夹下直接修改就行了，下面贴上图片输出结果如下图添加代码如下： x1=int(xyxy[0].item()) y1=int(xyxy[1].item()) x2=int(xyxy[2].item())

2021-10-15 12:04:03 36506 86

原创 python-批量图片格式转换

import osfrom PIL import Imagedirname_read="D:/wechatfile/Cam1_ImageFiles/"#这里换成你自己的文件位置dirname_write="D:/wechatfile/Cam1_ImageFiles_jpg/"#这里是保存位置names=os.listdir(dirname_read)#获取所有图片名字，并返回列表count=0for name in names: img=Image.open(dirname_read+

2021-10-14 17:28:53 255

原创标签制作（待更改）

```xml<?xml version="1.0"?>-<annotation><folder>Cam1_ImageFiles</folder><filename>2.png</filename><path>D:/wechatfile/Cam1_ImageFiles/2.png</path>-<source><database>Unknown</datab

2021-10-12 14:48:06 101

原创 pytorch resnet18网络动物分类完整的模型训练、验证、测试过程

先重写resnet18的模型，其实不用写也行，可以直接在models中调用import torchimport torch.nn as nnfrom torch.nn import functional as Fclass RestNetBasicBlock(nn.Module): def __init__(self, in_channels, out_channels, stride): super(RestNetBasicBlock, self).__init__()

2021-10-11 12:01:44 5268 23

原创修改文件夹下图片的名称

import ospath = ‘D:\pythonProject3\data’filelist = os.listdir(path)c=0for item in filelist:# print('item name is ',item)name = item.split('.')[0] + '.' + item.split('.')[1]#把图片的名字和编号分开后重新组合a=item.split('.')[0]b=item.split('.')[1]b=cc+=1b=str(b)

2021-10-10 21:15:15 169

原创报num_samples should be a positive integer value, but got num_samples=0错误

train=DataLoader(mydata_train,batch_size=64,shuffle=True)上面是对数据进行打包的一个片段，但是在对数据进行批量打包时出现以下的错误Traceback (most recent call last): File "D:/project/learn/aa.py", line 81, in <module> main() File "D:/project/learn/aa.py", line 76, in main t

2021-10-09 15:42:27 2428 7

原创 pytorch-复习与总结4

今天复习啥了呢？让我想想：大概是模型的创建Module和模型容器Contains模型创建首先来说一下模型创建，模型创建大概分为两大要素。1、构建子模块2、拼接子模块有哪些子模块呢？估计大家都见过，比如：卷积、池化、全连接怎么构建呢？怎么拼接呢？首先我们来复习一下怎么构建的问题：子模块的构建是你搭建类的__init__里面完成的，你构建的这个类一定要继承父类nn.Module。先来介绍一下torch.nn有哪些子模块1、nn.parameter 张量子类，表示可以学习的参数，w,b2、

2021-09-21 20:10:21 135

原创 pytorch复习与总结

今天来复习pytorch的数据读取机制torch.utils.data.DataLoader();构建可迭代的数据装载器，每一个for 循环，每一个iteration，都是从DataLoader中获取一个Batch_size大小的数据。有没有好奇过，就加载这几个类，然后就可以把数据读取，而且还能以批量的形式加载，这是怎样的一个过程呢？今天我们就来慢慢的深入学习，学到哪是哪。其中DataLoader大概有几个重要的参数，分别为：1、dataset：属于DataSet类，决定数据从哪读取，怎么读取。

2021-09-20 21:53:52 793 1

原创 pytorch复习总结与例子2

写在前面：学习pytorch也有一段时间了，一直都是在网上看别人的代码然后自己再敲一遍，把不理解的地方查找一下资料，但是总感觉缺点是什么，仔细思考一下整个过程，有点类似背代码的意思，这是非常要命的一点，所以今天就索性把这个在从头一点一点的利用晚上的时间再复习一边.今天我们复习的是计算图与动态机制深度学习一直在操控的是各种张量，少量的张量我们可以处理，但是随着网络的加深，各种参数的增多，张量的增多，会导致各种各样的问题产生，那么我们应该怎么处理这个现象呢？这时候便有了计算图的产生，那么什么又是计算图呢，

2021-09-17 20:23:18 164

原创 pytorch-复习与例子

写在前面：学习pytorch也有一段时间了，一直都是在网上看别人的代码然后自己再敲一遍，把不理解的地方查找一下资料，但是总感觉缺点是什么，仔细思考一下整个过程，有点类似背代码的意思，这是非常要命的一点，所以今天就索性把这个在从头一点一点的利用晚上的时间再复习一边，我们先来创建一个简单的线性回归方程，y=wx+b，这是一个一元一次方程，有点基础就知道，，神经网络是一个求参数的过程，通过不断的寻找最优解，不断的寻找参数。我们要知道这个网络创建的一个过程，首先应该是干什么，我们的任务是什么，y=wx+b中

2021-09-16 21:32:38 140

空空如也

空空如也