Ai_Taoism-CSDN博客

原创 pytorch的Backward过程用时太长？

pytorch的Backward过程用时太长?问题描述解决方案问题描述使用pytorch对网络进行训练的时候遇到一个问题，forward阶段很快(只需要几毫秒)，backward阶段却用时很长(需要十多秒)。导致这个问题的原因很容易被大家忽视，而且网上基本上没有直接的解决方案，经过一天的折腾，总算把导致这个问题的原因搞清楚了。解决方案导致这个问题的原因在于训练数据的浅拷贝，由于backward过程中的梯度是和模型推理过程中的张量相关的，如果这些张量在被模型使用之前没有被深拷贝，意味着backward

2022-05-19 15:59:59 2754 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 1434 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 3112 2

原创用pytorch计算神经网络模型需要占用的显存

用pytorch计算神经网络模型需要占用的显存#--------------------------------------------------------------------##作用：计算模型需要占用的显存，方便知道显卡够不够用#使用方法：将模型初始化之后，传入Calculate_gpu_memory()即可#--------------------------------------------------------------------#import torchimport

2021-09-28 19:46:23 2118

原创 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 4702 8

原创 CenterNet2的深入浅出（CVPR2021）

CenterNet2的深入浅出（CVPR2021）论文地址与代码主要贡献贡献解释论文地址与代码论文: https://arxiv.org/abs/2103.07461.代码: https://github.com/xingyizhou/CenterNet2.主要贡献CenterNet2是2021年的一篇CVPR论文，这篇论文读起来比较费解，因为里面牵涉了比较深的数学知识，以至于网上也没有比较好的解说，不过还好，在经过了一系列的资料查找之后，我将尽我最大的能力将下面写得深入浅出一点。该论文的主要贡

2021-08-06 22:20:46 1237 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 504

原创 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 491

原创 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 1373

原创 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 1941 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 826 5

原创 Pyinstaller踩坑记之将PPOCRLabel转成exe

Pyinstaller踩坑记一级目录二级目录三级目录一级目录二级目录三级目录

2021-07-01 10:31:49 1668 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 2192 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 2350 13

原创如何用pytorch选择性加载神经网络的权值

如何用pytorch选择性加载神经网络的相应层权值前言步骤前言步骤

2021-01-11 17:52:53 1037 1

原创一种可以加速深度学习推理的目标跟踪算法

自己发明的超实用目标检测算法前言算法原理算法实现前言所谓目标检测，就是在视频流的第一帧里面找到目标，标记目标，然后在后面的每一帧都准确找到目标的位置。算法原理算法实现

2020-12-30 23:26:39 394

原创 pytorch建立mobilenetV3-ssd网络并进行训练与预测

pytorch建立mobilenetV3-ssd网络并进行训练与预测一级目录二级目录三级目录一级目录二级目录三级目录

2020-12-29 11:32:49 3615 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 22198 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 3435 11

原创 C++调用简单的python程序

C++调用简单的python程序一、基本环境的搭建二、直接在C++里面调用执行python语句三、调用python脚本文件里面的定义函数调用不含参数的函数调用含一个参数的函数调用含多个参数的函数一、基本环境的搭建首先，用vs创建一个win32的控制台应用程序项目(vs2017及以上怎么创建百度一下就知道了)。然后配置好python的环境变量（把include文件夹加到包含目录，libs文件夹加到库目录，最好还把include文件夹加到附加包含目录）。如下图所示：最后，把python36.libs文

2020-12-09 11:19:41 2654 5

ahhhhhh520的博客