自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+

木盏

You only live once.

  • 博客(8)
  • 资源 (6)
  • 论坛 (2)
  • 收藏
  • 关注

原创 从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 216 2

原创 【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 165 1

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

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

2021-01-21 17:06:04 504 5

原创 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 518 1

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

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

2021-01-12 19:20:01 47

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

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

2021-01-11 16:04:08 302 1

原创 【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 349 1

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

CVPR2018模板latex

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

2020-12-02

FCOS检测算法训练模型

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

2020-11-19

inception_dec_2015

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

2017-11-18

CVPR2018论文_oral_2

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

2018-06-25

graphviz-2.38

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

2018-08-05

CVPR2018论文_oral

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

2018-06-25

木盏的留言板

发表于 2020-01-02 最后回复 2020-03-17

CSDN排名怎么搞的,我好几个月没进名次了

发表于 2018-08-17 最后回复 2018-08-17

空空如也

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

TA关注的人 TA的粉丝

提示
确定要删除当前文章?
取消 删除