2021年01月_木盏

12月 11月 10月 09月 08月 07月 06月 05月 04月 03月 02月 01月

原创从Pytorch模型pth文件中读取参数成numpy矩阵

目的：把训练好的pth模型参数提取出来，然后用其他方式部署到边缘设备。Pytorch给了很方便的读取参数接口：nn.Module.parameters()直接看demo：from torchvision.models.alexnet import alexnetmodel = alexnet(pretrained=True).eval().cuda()parameters = model.parameters()for p in parameters: numpy_para

2021-01-24 03:43:22 4542 5

原创【torchsummary】Pytorch模型参数统计工具

今天介绍的工具是torchsummary，可以用来统计PyTorch每层的参数情况。一来可以用于参数剪枝优化，二来可以用于了解模型的参数分布。安装：pip install torchsummary使用：from torchvision.models.alexnet import alexnetfrom torchsummary import summarymodel = alexnet(pretrained=True).eval().cuda()summary(model, i

2021-01-24 03:07:32 3688 1

原创使用TensorRT加速Pytorch模型(简单实用)

TensorRT是Nvidia官方给的推理加速工具，如同OpenVINO之于Intel。支持诸多的AI框架，如Pytorch，caffe，MXnet，tensorflow等。此外还对某些热门框架有特别的照顾，比如针对PyTorch有直接转换的工具torch2trt（咱们一会儿说）。

2021-01-21 17:06:04 17090 22

原创 Pytorch模型转成onnx并验证结果(排坑经验帖)

很多时候有pytorch模型转onnx模型的必要，比如用tensorRT加速的时候。本文将介绍pytorch的pth模型如何转换成onnx，并且验证你转的模型对不对。先给官网链接：https://pytorch.org/docs/stable/onnx.html咱们直接用一段代码来看：（本人亲自整理，有问题可留言交流~）import os.path as ospimport numpy as npimport onnximport onnxruntime as ortimport t

2021-01-14 22:38:43 12520 9

原创从内存表现上简单分析Python中的拷贝

数据拷贝是Python较为容易出错的地方。本文针对等号复制、浅拷贝、深拷贝和可变对象、不可变对象来一一讲解。1. 等号复制在一般的编程语言中，习惯用“=”表示复制或拷贝。但这个在Python里面要特别注意。Python的等号只是传递变量的引用而已。哪怕只是传一个int型的变量，Python传的也只是引用(类似于C/C++的指针)，只是在计算过程中会把结果用额外的空间存储，所以对于数值类型的变量，我们可以用等号复制。我们可以用id()来查看变量的地址：import ctypesa = 1b

2021-01-12 19:20:01 312

原创【半精度】Pytorch模型加速和减少显存

如标题所示，这是PyTorch框架提供的一个方便好用的trick：开启半精度。直接可以加快运行速度、减少GPU占用，并且只有不明显的accuracy损失。

2021-01-11 16:04:08 31448 11

原创【Python可视化】matplotlib画动态曲线

matplotlib是python里最popular的画图工具，最近发现它可以用来画动态图。可以看看AI模型训练时候的loss曲线，accuracy变化等等。客观上，比tensorboard更简洁。代码还是很简单滴：import matplotlib.pyplot as pltfrom random import randomdef do_something(): for p in range(10000000): res += pfig, ax = plt.su

2021-01-05 20:35:21 10174 9

原创 linux终端分割工具tmux

tmux可以把linux下的terminal分割成用户想要的样子，可以方便观测。1. 安装方法：(ubuntu下)sudo apt-get install tmux2. 快速使用方法tmux安装好以后，直接在终端输入tmux，就可以进入tmux会话。然后就可以分割窗口了。上下分割：tmux split-window左右分割：tmux split-window -h调整分割比例：左手按住ctrl+b不动，右手按方向键(上、下、左、右)。即可完成相应调整。 ..

2021-01-04 15:46:08 1350

CVPR2018模板latex

这是CVPR论文的latex模板，由于国内访问overleaf经常不稳定、网速也慢，而且需要注册登录十分麻烦。于是我传到CSDN上方便大家交流学习使用。

2020-12-02

FCOS检测算法训练模型

FCOS检测算法在VOC数据集上的训练模型，直接下载可进行inference。配合代码https://github.com/leviome/fcos_pure 使用。具体教程看README.

2020-11-19

graphviz-2.38

在网上找到的安全无毒的graphviz-2.38，象征性收1分下载分

2018-08-05

CVPR2018论文_oral_2

CVPR2018的oral论文合集，Oral一共有68篇，受上传文件大小限制，分成part1和part2压缩文件。Oral论文是CVPR中最优质的论文，best paper/honorable mention什么的都在oral里面。我花了很多工夫整理出来，来分享给各位CVer。还有少数几篇论文暂时没公开，所以只找出64篇。还有一点要说明：该文档仅作交流学习用处，不能当作商业用途。

2018-06-25

inception_dec_2015

这是来自https://storage.googleapis.com/download.tensorflow.org的模型文件，经常链接失效，或者下载速度奇慢。我花了很大功夫下载好，供大家使用。

2017-11-18

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人