- 博客(24)
- 资源 (3)
- 收藏
- 关注
原创 pytorch的Backward过程用时太长?
pytorch的Backward过程用时太长?问题描述解决方案问题描述使用pytorch对网络进行训练的时候遇到一个问题,forward阶段很快(只需要几毫秒),backward阶段却用时很长(需要十多秒)。导致这个问题的原因很容易被大家忽视,而且网上基本上没有直接的解决方案,经过一天的折腾,总算把导致这个问题的原因搞清楚了。解决方案导致这个问题的原因在于训练数据的浅拷贝,由于backward过程中的梯度是和模型推理过程中的张量相关的,如果这些张量在被模型使用之前没有被深拷贝,意味着backward
2022-05-19 15:59:59 3358 7
原创 pytorch中forwod函数在父类中的调用方式
pytorch中forwod函数在父类中的调用方式问题背景问题背景最近在研究Detetron2的代码结构时,发现有些网络代码里面没有forward函数,却照样可以推理,深入挖掘之后,发现其将forword函数都写在了同一个父类里面。这就牵涉到了下面这个问题,子类中没有forward函数,只有父类中有forward函数,这样能不能正常调用网络。import torch.nn as nnclass Network1(nn.Module): def __init__(self):
2022-04-26 21:41:37 1498 1
原创 Detectron2和fvcore中的Registry机制详解
Detectron2和fvcore中的Registry机制详解为什么要有Registry机制如何使用Registry机制Registry机制内部是怎么实现的为什么要有Registry机制Registry机制原本是fvcore框架里面的功能,因为fvcore是一个轻量级的核心库,它提供了在FAIR开发的各种计算机视觉框架(如Detectron2、PySlowFast和ClassyVision)中共享的最常见和最基本的功能。所以Detectron2里面有很多功能都是直接从fvcore框架里面直接拿过来用的。
2022-04-24 12:54:51 3358 2
原创 用pytorch计算神经网络模型需要占用的显存
用pytorch计算神经网络模型需要占用的显存#--------------------------------------------------------------------##作用:计算模型需要占用的显存,方便知道显卡够不够用#使用方法:将模型初始化之后,传入Calculate_gpu_memory()即可#--------------------------------------------------------------------#import torchimport
2021-09-28 19:46:23 2288
原创 pytorch实现DETR的推理程序
pytorch实现DETR的推理程序为什么要写这个代码推理结果比较为什么要写这个官方代码在https://github.com/facebooksearch/detr,但是官方代码没有提供推理代码,只提供了训练和评估的代码,所以我在这里写一下推理代码。代码#这个文件是自己写的import argparseimport randomimport timefrom pathlib import Pathimport numpy as npimport torchfrom models im
2021-08-08 20:32:12 5342 10
原创 CenterNet2的深入浅出(CVPR2021)
CenterNet2的深入浅出(CVPR2021)论文地址与代码主要贡献贡献解释论文地址与代码论文: https://arxiv.org/abs/2103.07461.代码: https://github.com/xingyizhou/CenterNet2.主要贡献CenterNet2是2021年的一篇CVPR论文,这篇论文读起来比较费解,因为里面牵涉了比较深的数学知识,以至于网上也没有比较好的解说,不过还好,在经过了一系列的资料查找之后,我将尽我最大的能力将下面写得深入浅出一点。该论文的主要贡
2021-08-06 22:20:46 1440 2
原创 FCOS论文整理(CVPR2021)
FCOS论文整理(CVPR2021)论文地址和代码主要贡献贡献解释论文地址和代码论文: https://arxiv.org/pdf/1904.01355.pdf.代码: https://github.com/tianzhi0549/FCOS/.主要贡献这是2021年的一篇CVPR论文,该论文的主要贡献有如下几点:●定义了一套不使用锚框,只使用全卷积的Anchor-free目标检测算法(FCOS)。●利用金字塔模型(FPN)进行多尺度预测,解决了Anchor-free目标检测算法中的一个公认的缺
2021-08-06 15:22:52 540
原创 YOLOF论文整理(CVPR2021论文)
@TOC论文地址和代码论文: https://arxiv.org/abs/2103.09460.代码: https://github.com/megvii-model/YOLOF.主要贡献这是2021年的一篇CVPR论文,该论文的主要贡献有如下几点:●通过实验证明,金字塔模型(FPN)的主要作用不是多尺度特征融合,而是分而治之的解决思路。这一点目前几乎被整个行业误解,导致几乎所有的目标检测网络都把FPN加了进去。●提出单进单出编码器Dilated Encoder,并用这个编码器代替FPN,精度
2021-08-03 20:23:16 538
原创 CVPR2021论文翻译:Yang Interactive Self-Training With Mean Teachers for Semi-Supervised Object Detection
CVPR2021论文翻译:Yang Interactive Self-Training With Mean Teachers for Semi-Supervised Object DetectionAbstract(摘要)1.Introduction(介绍)2.Related Works(相关工作)3.Method(方法)3.1.Preliminary(准备工作)3.2.Pseudo Labels Fusion(伪标融合)3.3.Interactive Self-Training(互动自训练)3.4.Mea
2021-07-21 19:28:27 1553
原创 ubuntu的wps无法输入中文解决方案(2021年)
@[TOC](ubuntu wps无法输入中文解决方案(2021年))市面上的解决方案市面上的解决方案一般和下面这个链接一样:https://blog.csdn.net/sinat_36742186/article/details/88665701本文的解决方案其实上面的解决方案本没有错,但是它忽略了一个问题,不是所有的ubuntu电脑的键盘输入法系统都是fcitx,所以在加入环境变量的时候,就不能机械地用下面这一句。export XMODIFIERS="@im=fcitx"export QT
2021-07-19 22:10:52 2255 3
原创 用deepstream-test3跑yolov5,并对小汽车(car)进行流量统计
用deepstream-test3跑yolov5,并对小汽车car进行流量统计前言项目效果项目地址前言本文的项目是在Nvidia jetson上部署yolov5算法,并实现自定义功能(如对车辆进行流量统计)。在这里我是在deepstream-test3源码的基础上,对gstreamer进行更改,加入自定义功能,最后编译实现。项目效果见:https://www.bilibili.com/video/BV1Xh41187tQ/项目地址见:https://github.com/dongdongdong
2021-07-01 16:38:00 929 5
原创 Pyinstaller踩坑记之将PPOCRLabel转成exe
Pyinstaller踩坑记一级目录二级目录三级目录一级目录二级目录三级目录
2021-07-01 10:31:49 2059 9
原创 Jetson AGX Xavier 刷机教程
Jetson AGX Xavier 刷机教程前言step1:用虚拟机安装sdkmanagerstep2:用sdkmanager刷机前言首先,我们需要知道刷机的大致流程:先用一个带ubuntu系统的主机(可以是虚拟机,但不可以是win10自带的ubuntu子系统)安装好sdkmanager软件,然后用nvidia自带的usb线连接nvidia盒子与ubuntu主机,然后用sdkmanager刷机。在用虚拟机刷机的时候,需要注意的给虚拟机分配的内存空间必须足够大,至少要40G。step1:用虚拟机安装s
2021-04-02 11:02:25 2398 2
原创 paddleOCR踩坑记
paddleOCR踩坑记训练莫名终止或者预测莫名终止训练和评估的效果都特别好,预测的效果却特别差训练莫名终止或者预测莫名终止这是因为在tools/program.py文件里有一句逻辑关系语言出错了,需要进行如下修改。if idx >= len(train_dataloader): break#把上面这句话修改成下面这个if idx >= len(train_dataloader)-1: break注意,有两个地方都要这样改,train里面和eval里面。训练和评估的效果都特别
2021-02-26 11:49:32 2600 13
原创 一种可以加速深度学习推理的目标跟踪算法
自己发明的超实用目标检测算法前言算法原理算法实现前言所谓目标检测,就是在视频流的第一帧里面找到目标,标记目标,然后在后面的每一帧都准确找到目标的位置。算法原理算法实现
2020-12-30 23:26:39 445
原创 pytorch建立mobilenetV3-ssd网络并进行训练与预测
pytorch建立mobilenetV3-ssd网络并进行训练与预测一级目录二级目录三级目录一级目录二级目录三级目录
2020-12-29 11:32:49 3943 3
原创 如何将python程序打包成DLL
如何将python程序打包成DLL前言Step1:用cython生成python脚本的.h和.c文件Step2:生成dll文件Step3:测试dll文件前言将python程序打包成DLL文件,然后用C++调用生成的DLL文件,这是一种用C++调用python的方法,这一块比较容易遇到坑。网上关于这一块的教程不是很多,而且大部分都不能完全解决问题。我在傻傻挣扎了几天之后,终于试出了一个可行的版本,写在这里供大家参考,也是供以后的自己参考。Step1:用cython生成python脚本的.h和.c文件首
2020-12-12 16:43:45 23809 22
原创 tensorflow2.0建立MobilenetV3网络并进行训练与预测
tensorflow2.0建立MobilenetV3网络并进行训练与预测Step1:前言Step2:建立bneck卷积块激活函数轻量级注意力模型bneck的建立Step3:构建MobilenetV3_small网络Step4:如何调用建立的MobilenetV3进行训练训练数据的形式与预处理。训练数据的调用训练方式1:使用原始数据训练训练方式2:使用别人的预训练权值进行训练(迁移学习)Step5:如何利用训练好的MobilenetV3模型进行预测Step1:前言之所以采用tensorflow2.0来建立
2020-12-09 21:19:06 3714 11
原创 C++调用简单的python程序
C++调用简单的python程序一、基本环境的搭建二、直接在C++里面调用执行python语句三、调用python脚本文件里面的定义函数调用不含参数的函数调用含一个参数的函数调用含多个参数的函数一、基本环境的搭建首先,用vs创建一个win32的控制台应用程序项目(vs2017及以上怎么创建百度一下就知道了)。然后配置好python的环境变量(把include文件夹加到包含目录,libs文件夹加到库目录,最好还把include文件夹加到附加包含目录)。如下图所示:最后,把python36.libs文
2020-12-09 11:19:41 2782 5
原创 函数插值方法综述
函数插值方法综述引言为什么可以用多项式函数做插值常用的插值方法及matlab实现拉格朗日插值牛顿插值埃尔米特插值分段低次插值三次样条插值引言为什么可以用多项式函数做插值常用的插值方法及matlab实现拉格朗日插值牛顿插值埃尔米特插值分段低次插值三次样条插值...
2020-06-25 12:50:00 3074
原创 两行代码爬取股票的全部信息
目录先看效果代码讲解第一行代码第二行代码先看效果下图展示的是我用python爬取的英维克股票(编号是002837)的日k数据,是3年内的所有日k信息哦,而且最新的数据可以到最近一个交易日结束哦。数据解释:date:日期open:开盘价high:最高价close:收盘价low:最低价volume:成交量price_change:价格变动p_change:涨跌幅ma5:5日均价ma10:10日均价ma20:20日均价v_ma5:5日均量v_ma10:10日均量v_ma20:2
2020-05-27 13:40:59 1371
原创 基于BP神经网络的室内声源定位算法(附有程序)
基于BP神经网络的室内声源定位算法的实现(附有程序)问题描述现在有一个安静的房子,有一个人在房间里走动,我要利用屋里的麦克风接收这个人的脚步声,然后对这个人进行定位。问题的意义声源定位,这个问题的研究意义重大,它能克服视觉定位的缺点(即只能对看得到的地方进行定位)。问题的研究方法本文只讨论基于麦克风阵列的声源定位(即利用麦克风收集声源信息)。目前解决这个问题的主流方法有三个,分别是基于最大输出功率的可控波束形成技术、基于高分辨率谱估计技术、基于声达时间差的定位技术。这三种方法都是通过研究声音的
2020-05-18 21:45:38 9576 17
原创 matlab里det函数的神奇bug
matlab里det函数的神奇bug最近在使用matlab的det函数算两个矩阵的乘积的行列式时候,发现了一个神奇的bug,先看下面这个计算:由高等代数的知识可以知道,det(ab)=det(ba)才对,但是从上面的结果可以看出,det(ab)不等于det(ba),起码它们之间存在误差,如果你觉得这个误差是可以忽略的,那你接下来往下看。有没有发现误差从10^(-12) 变成了10^(-9),如果你觉得这也能忽略不计,那你接下来再往下看。这个误差已经达到了10^(4),你觉得还可以忽略吗?当
2020-05-12 00:41:08 9214 3
基于BP神经网络算法的室内声源定位.zip
2020-05-18
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人