PyTorch Notes
Pytorch
血_影
关注:计算机视觉、图形图像处理、机器学习、人工智能等领域.
喜欢:C/C++/python
邮箱:1013279512@qq.com
展开
-
Pytorch常见问题
Can‘t get attribute ‘SiLU‘ on <module ‘torch.nn.modules.activationubuntu 18.04 serverpython3.8.0pytorch==1.6.0 torchvision==0.7.0解决方法将SiLU拷贝到~/.local/lib/python3.8/site-packages/torch/nn/modules/activation.py文件中class SiLU(Module): # export-friendly原创 2021-05-06 15:46:33 · 855 阅读 · 2 评论 -
TVM问题集
Auto-tuning 报错参考代码:https://github.com/gary30404/tvm-yolov3.git运行一会之后程序报错,出现ImportError: cannot import name ‘aggcv’解决办法是将tvm/autotvm/tuner/xgboostcost_model 450行修改为try: from xgboost.training import aggcvexcept ImportError: from xgboost.callba原创 2021-04-19 17:28:13 · 428 阅读 · 0 评论 -
实用笔记系列5
ABCNet Bezier Align模块编译在做场景文字识别时,经常需要对任意形状的文本进行检测识别,这一直以来是场景文字识别的一个难点,也不断有新的算法出现来解决这个问题。近期的如ABCNet就是众多方法之一。它主要采用参数化的贝塞尔曲线来自适应地处理任意形状的文本并增加了一层bazier align来提取任意形状文本实例的精确卷积特征。其算法框架如下图所示:详细介绍可以参考网上相关教程,其中三次贝塞尔曲线拟合图如下所示:当我们已经有一个算法模型能够检测出文本框bbox和详细的8个点的时候就可原创 2020-08-28 17:16:57 · 430 阅读 · 0 评论 -
实用笔记系列3
Labelme 转cocoLabelme 标注图像生成的json格式{ "version": "4.2.10", "flags": {}, "shapes": [# 每个对象的形状 { # 第一个对象 "label": "malignant", "line_color": null, "fill_color": null, "...原创 2020-04-17 17:00:41 · 243 阅读 · 0 评论 -
VO数据集详解
Visual Object Classes Challenge 2012 (VOC2012)IntroductionThe main goal of this challenge is to recognize objects from a number of visual object classes in realistic scenes (i.e. not pre-segmented o...原创 2019-10-23 17:52:44 · 1030 阅读 · 1 评论 -
Pytorch转onnx、torchscript
前言 本文将介绍如何使用ONNX将PyTorch中训练好的模型(.pt、.pth)型转换为ONNX格式,然后将其加载到Caffe2中。需要安装好onnx和Caffe2。PyTorch及ONNX环境准备为了正常运行ONNX,我们需要安装最新的Pytorch,你可以选择源码安装:git clone...原创 2019-07-28 12:18:04 · 12826 阅读 · 6 评论 -
pytorch 之 构建网络模型的几种方法
利用pytorch来构建网络模型有很多种方法,以下简单列出其中的四种。假设构建一个网络模型如下:卷积层--》Relu层--》池化层--》全连接层--》Relu层--》全连接层首先导入几种方法用到的包:import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd impo...原创 2019-03-04 11:20:18 · 574 阅读 · 0 评论 -
pytorch实用笔记系列1
Finetune权值初始化torch.save(pre_net.state_dict(), 'pretrained.pkl') # save a pretrained modelpretrained_dict = torch.load('pretrained.pkl') # load mode parammodel = Net() # create a new modeln...原创 2019-09-04 20:24:26 · 352 阅读 · 0 评论 -
FaceLandmark dataset Augment
前言在调研人脸关键点检测算法时,找到了一篇较新的人脸106个点检测的论文Grand Challenge of 106-Point Facial Landmark Localization, 进一步深挖该篇论文,发现新构建的人脸关键点数据集----JD- landmark,并且已经开源了,于是乎,立马给项目负责人Dr. Hailin Shi发了封希望能够获得一份该数据集的邮件,数据集就这么快速的到...原创 2019-09-01 12:46:23 · 2405 阅读 · 6 评论 -
Pytorch 如何高效使用GPU
前言深度学习涉及很多向量或多矩阵运算,如矩阵相乘、矩阵相加、矩阵-向量乘法等。深层模型的算法,如BP,Auto-Encoder,CNN等,都可以写成矩阵运算的形式,无须写成循环运算。然而,在单核CPU上执行时,矩阵运算会被展开成循环的形式,本质上还是串行执行。GPU(Graphic Process Units,图形处理器)的众核体系结构包含几千个流处理器,可将矩阵运算并行化执行,大幅缩短计算时间...转载 2019-07-31 22:59:02 · 22565 阅读 · 6 评论 -
PyTorch - Python deep learning neural network API
Welcome back to this series on neural network programming with PyTorch. To kick this series off, let’s introduce PyTorch, a deep learning neural network package for Python. There’s no better place to ...转载 2019-09-11 09:54:42 · 557 阅读 · 0 评论