- 博客(44)
- 收藏
- 关注
RSS订阅原创 Pytorch与Onnx的转换与推理
首先要强调的是，有两个版本的onnxruntime，一个叫onnxruntime，只能使用cpu推理，另一个叫onnxruntime-gpu，既可以使用gpu，也可以使用cpu。.pt表示pytorch的模型，.onnx表示onnx的模型,后缀名为.pt, .pth, .pkl的pytorch模型文件之间其实没有任何区别。有两种方式可用于保存/加载pytorch模型 1）文件中保存模型结构和权重参数 2）文件只保留模型权重.二、pytorch模型转ONNX模型。1、文件中保存模型结构和权重参数。
2023-10-18 21:54:10
881
原创 windows平台FairMOT的实现
5.不要卸载setuptools，在安装dcn_v2时，出现错误，根据搜索提示，需要安装低版本的setuptools，我于是卸载了setuptools导致每次运行都会报一个文件出错，于是还重新安装了annaconda。（遇见flat_hash_map.h错误，没解决，但是通过重安pytorch方法，版本匹配后，就没有这个错误了）1.torch1.1.0+torchvision0.3.0+cuda9.2+vs2015可以成功编译dcn_v2，但是编译apex时候报错，参照。一定要选择上C++编译工具。
2023-10-09 22:29:36
534
原创 Homography单应性矩阵
比如 上图中的 桌面，地面，橱柜面。对于这种图像中有多个平面的情况，我们就需要针对每一个平面使用其对应的Homography了。对于所有的对应点，只要它们都位于同一个物理平面上，上述 Homography 就是成立的。换句话说，就是可以把图一中书皮上的所有点都映射到图二的书皮上，也就是看起来，图一中的书皮和图二中的书皮对齐了！考虑 同一个平面(比如书皮)的两张图片，红点表示同一个物理坐标点在两张图片上的各自位置。将矩阵带入变换公式，可以得到实际的变换为：x' = w-x, y' = y。
2023-07-22 21:19:27
158
原创 Pytorch出现RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)
上面四行任选一，还有其他未列出的表述方法，都可以将模型加载到GPU。有时已经写了将模型和输入数据加载到GPU的语句，但还是报了这个错误。问题原因搞清楚了，模型没加载到CPU，在代码中加一行语句就可以了。这个函数功能是产生一个新的tensor，并不会改变原数据。问题来源是输入数据没有加载到GPU，解决方法为。是一个“in-place”方法，从报错问题描述中可以找到错误原因。
2023-07-18 18:49:30
392
1
原创 RuntimeError: xxx.pth is a zip archive (did you mean to use torch.jit.load()?)
RuntimeError: did you mean to use torch.jit.load()?
2022-08-31 11:51:56
1902
1
原创 加载ECANet参数
文章：ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks因为最近想用这个文章的模型，但是不想把整个模型的参数加载到我的网络，想要把layer1和layer2等模块的参数逐个加载进我的模型中，因此苦苦摸索了一个比较笨的方法，大神勿喷。首先先查看ECA-Net中各个层的编号：import torchres_eca=torch.load('./eca_resnet50_k3557.pth.tar')f
2022-03-26 18:34:58
404
原创 pytorch batch size
当数据集的图片大小不一致时，设置batchsize>1，报如下错误：epoch 0, processed 0 samples, lr 0.0000001000Traceback (most recent call last): File "train.py", line 347, in <module> main() File "train.py", line 138, in main train(train_list, model, cri
2021-07-10 20:17:59
1186
原创 Could not resolve proxy - git clone
参考：https://stackoverflow.com/questions/50996912/could-not-resolve-proxy-git-clone/51016837Git has its own proxy.To reset git proxy:git config --global https.proxy ""git config --global http.proxy ""To reset system proxy:On Ubuntu, you can set p
2020-10-30 14:49:13
1872
原创 windows10+anaconda+cuda9.2+cudnn7+pytorch+tensorflow
4.pytorch安装：采用先将安装包下载下来再安装的方法首先根据自己的python、cuda、cudnn版本来选择下载哪个版本，清华源：https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/下载后放在哪里都可以。然后用conda安装（我用pip安装总是不成功）。D:\anzhuang>conda install pytorch-1.5.1-py3.8_cuda92_cudnn7_0.tar.bz2.
2020-08-04 11:25:17
802
原创 解决apt-get install E: 无法定位软件包问题和没有可安装候选软件包
参考:https://www.jianshu.com/p/b93ff6aa4022一 无法定位软件包1 编辑sources.list$ sudo vim /etc/apt/sources.list2 添加源在sources.list文件末尾添加deb http://archive.ubuntu.com/ubuntu/ trusty main universe restricted multiverse3 更新软件列表$ sudo apt-get update二没有可安装
2020-08-02 18:18:08
24554
12
原创 ubuntu安装VisualStdioCode
sudo add-apt-repository ppa:ubuntu-desktop/ubuntu-makesudo apt-get updatesudo apt-get install ubuntu-make sudo umake ide visual-studio-code
2020-07-16 11:04:54
349
1
原创 E: 无法定位软件包 gedit-gmate
要安装GMate，你首先需要添加正确的PPA。打开终端，运行下面的命令：sudo apt-add-repository ppa:ubuntu-on-rails/ppasudo apt-get update完成后。就可以运行：sudo apt-get install gedit-gmate
2020-07-16 10:44:32
3918
1
原创 ubuntu禁止自动更新内核
总结了三处需要改的地方。1.查看自己电脑的内核dpkg --get-selections | grep linux得到如下信息：查看正在使用的内核uname -a得到如下信息：禁止更新内核：sudo apt-mark hold linux-image-4.13.0-36-genericsudo apt-mark hold linux-image-extra-4.13.0-36-genericsudo apt-mark hold linux-headers-4.1
2020-07-15 21:47:45
13995
原创 cuda10.1+cudnn7.6.5
1、CUDA 10.1安装（包含Nvidia显卡驱动）如果Ubuntu系统下没有装过Nvidia显卡驱动，不必单独安装，直接装CUDA即可自动安装显卡驱动.（自己手动安装的显卡驱动很有可能与cuda冲突，cuda10.1推荐安装的驱动版本是418.67）（1）先 Ctrl+Alt+F1 进入命令行模式：sudo service lightdm stopsudo bash cuda_10.2.89_440.33.01_linux.run按照它要求的选择accept，install，耐心等它
2020-07-15 16:04:17
3232
2
原创 Ubuntu apt-get update 更新失败：无法认证来自该源的数据，所以使用它会带来潜在风险
参考博客：https://blog.csdn.net/q290994/article/details/80577711ubuntu下运行sudo apt-get update出现如下错误：W: 仓库 “http://mirrors.aliyun.com/ubuntu xenial Release” 没有 Release 文件。N: 无法认证来自该源的数据，所以使用它会带来潜在风险。N: 参见 apt-secure(8) 手册以了解仓库创建和用户配置方面的细节。W: 仓库 “http://mi
2020-07-14 17:03:28
4175
原创 安装ubuntu教程
一.首先做启动盘使用软碟通（https://cn.ultraiso.net/xiazai.html）或者Rufus（http://rufus.ie/zh_CN.html）以下安装教程参考：https://blog.csdn.net/hitzijiyingcai/article/details/81627816以软碟通为例：1.下载安装完成之后打开软件2.将iso文件加载完以后点开启动下的写入硬盘硬盘映像3.然后选择我们自己的U盘（可以提前格式化），再写入就行了完...
2020-07-14 12:19:34
3872
2
原创 ssh的一些操作
1.查看ssh状态:ps -e | grep ssh出现: 1027 ? 00:00:00 sshd说明已启动,若只有agent,说明没有启动,可自行手动启动2.启动或关闭ssh服务sudo /etc/init.d/ssh startsudo /etc/init.d/ssh stop3.查看对方电脑ipifconfig例如:enp4s0 Link encap:Ethernet HWaddr 30:9c:23:15:4b:61
2020-07-12 17:32:33
234
原创 ubuntu下查看电脑硬件的基本信息
1.查看cpu信息cat /proc/cpuinfo2.查看内存cat /proc/meminfo3.查看硬盘sudo fdisk -l | grep "Disk /dev/sd"
2020-05-30 11:18:48
2448
原创 matlab操作h5文件
1.显示.h5文件的详细信息h5disp('IMG_1.h5')得到如下结果:HDF5 IMG_1.h5 Group '/' Dataset 'density' Size: 1024x768 MaxSize: 1024x768 Datatype: H5T_IEEE_F32LE (single) ChunkSize: [] Filters: none FillValue:
2020-05-09 11:27:44
2511
原创 scipy安装慢
1、pip install scipy -i https://pypi.douban.com/simple/2、pip install --index https://pypi.mirrors.ustc.edu.cn/simple/ + (要下载的模块)pandas3、pip install --index http://mirrors.sohu.com/python/例如...
2020-02-14 16:35:13
5659
1
原创 pytorch model参数解析
以vggnet16为例:mod = models.vgg16(pretrained = True)ps:在pytorch中，torch.nn.Module模块中的state_dict变量存放训练过程中需要学习的权重和偏执系数，state_dict作为python的字典对象将每一层的参数映射成tensor张量，1.print("mod:",mod)输出:('mod:', VG...
2020-01-07 15:34:14
2039
原创 /boot空间不足的解决办法
1.查看系统已经安装的内核版本dpkg --get-selections |grep linux-image2.查看系统当前使用的内核版本uname -a3.常规利用删除旧版本内核的方法来解决该问题，但是总会遇到如下问题：有未能满足的依赖关系。请尝试不指明软件包的名字来运行“apt-get -f install”(也可以指定一个解决办法)。于是执行命令apt-...
2019-11-04 18:15:45
443
原创 ubuntu下安装opencv
1.在官网上下载opencv：http://opencv.org/releases.html2.解压zip包unzip opencv-3.4.3.zipcd opencv-3.4.33.安装依赖库和cmakesudo apt-get install cmake sudo apt-get install build-essential libgtk2.0-dev liba...
2019-11-04 18:02:11
372
原创 skimage图像的读取与保存
首先,说明用opencv与skimage.io.imread读取和保存图片的区别:读取和保存后的都是numpy格式，但cv2的读取和存储格式是BGR，而skimage的读取和存储格式是RGB。1.读取图片:#skimage读取图片img=skimage.io.imread(file_dir, as_grey=False)#imread读取图片,0:grey,1:color,默认...
2019-07-25 16:54:41
12481
原创 ccnn
1.ImportError: no module named vigra解决:参考:https://anaconda.org/conda-forge/vigraconda install -c conda-forge vigra使用如上命令，利用anaconda2安装vigra：会使得conda自动安装python3.7，覆盖掉python2.7，从而导致错误：ImportEr...
2019-07-24 17:58:13
2718
原创 ubuntu下安装theano
1.conda install theano2.提示no model name lasagne解决:(1)在github上下载Lasagne(2)因为前面已经安装过theano了,因此直接安装python setup.py install3.提示no model named setuptoolsconda install setuptools...
2019-06-23 21:13:26
231
原创 pytorch安装
安装pytorch之前先查看ubuntu下cuda的版本号，再进行安装conda install pytorch torchvision cudatoolkit=8.0 -c pytorch如果安装速度慢的话，去掉-c pytorch。因此“-c pytorch”默认从pytorch官网下载安装包，因此安装的时候将-c pytorch去掉，再修改为清华源，可提高安装速度。指定pyt...
2019-05-30 11:51:48
913
原创 ｐytorch运行程序过程中遇到的问题
将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告，它在提醒用户下个版本这将成为一个错误错误１：UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python...
2019-05-30 11:42:44
1652
原创 安装驱动时的问题
1.If you plan to no longer use the NVIDIA deriver, you should make sure that no X screens are configured to use the NVIDIA X driver X driver in your X configuration file. If you used nvidia-xconfig to...
2019-05-15 10:32:21
5277
原创 sift-flow 数据集
sift-flow数据集解析：1.semanticlabels:语义标签。每一个像素点用一种类别来表示，每一种类别用一个数字来表示。总共有33类，背景类用0表示。对应列表：01 awning 雨蓬02 balcony 阳台03 bird 鸟04 boat 船05 bridge 桥06 building 建筑07 bus 大巴车08 car 小汽车09 cow 牛10 cr...
2019-05-09 11:20:18
2768
2
原创 caffe撸码：solver.cpp
template <typename Dtype>void Solver<Dtype>::Step(int iters) { const int start_iter = iter_; const int stop_iter = iter_ + iters; int average_loss = this->param_.average_loss();...
2019-05-07 17:12:57
1465
原创 caffe撸码：euclidean.cpp
namespace caffe {template <typename Dtype>void EuclideanLossNormLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, const vector<Blob<Dtype>*>...
2019-05-07 16:46:07
230
原创 终止正在运行的程序
Ubuntu下运行程序，如果按快捷键ctrl+c无法终止程序，可以按照如下步骤终止程序：nvidia-smi假如我们想关掉正在运行的caffe程序，看一下它的PID编号是21457，则执行以下命令即可终止程序kill -9 21457 ...
2019-01-19 10:41:14
6245
原创 protobuf的一些操作
1.查看protobuf版本protoc --version ps:假如ubuntu有两个版本的protobuf，版本3.1.0安装在usr/local/bin下，版本3.0.0安装在$HOME/anaconda2/bin下，sudo protoc --version 会显示当前版本是3.1.0，路径指向的是usr/local/binprotoc --ver...
2019-01-19 10:29:28
2204
原创 ubuntu16.04+anaconda2+opencv3.4.0
1.编译好的opencv和自己编译安装opencv的区别Ubuntu安装了anaconda以后，$conda install opencv直接安装网上编译好的opencv，但是Ubuntu，python以及opencv不同版本之间使用都有差异，容易出问题；但是如果没出问题的话，这确实是最简单的安装方式.$pip install opencv-python或者$apt-get i...
2018-12-02 20:24:04
1041
原创 降级gcc版本
降级安装gcc/g++版本为4.7.x$ sudo apt-get install -y gcc-4.7 g++-4.7等下载安装完成之后，我们可以去/usr/bin/目录下看到一些gcc开头的文件，其中有一个是gcc，他是个链接文件，链接到当前版本的gcc文件（例如我的电脑就是gcc-5），也即是说，他是链接到当前使用的gcc上的，所以改变他的链接源文件就可以了，也就是让gcc这个链接文...
2018-11-13 21:43:14
3943
原创 /boot 100%
wanglin@wanglin-System-Product-Name:~/桌面$ df发现 /boot 100%出现这个问题的原因是系统的冗余内核太多。此时在终端下输入 再输入uname -a显示当前正在使用的系统内核发现正在使用的系统内核为4.15.0-36，因此需要卸掉冗余的内核。如果内核的状态是install，说明：系统已经安装了相应的内核，使用purge命令删除...
2018-11-13 21:33:36
590
原创 python保存图片
原图如下： 1. 保存图片对象img_s.save(os.path.join("/home/wanglin/PycharmProjects/data_test/data/Train400/original.jpg")) 2.保存图片numpy数组，但是会改变图片的颜色cv2.imwrite(os...
2018-11-05 18:11:17
13995
3
原创 ubuntu16.04之前能上网，突然有线和无线网都用不成了
苦苦折腾了好几天，试了A方法：https://jingyan.baidu.com/article/7f766daf42b7984101e1d0be.htmlB方法：C方法：http://blog.csdn.net/ifmvo/article/details/54023628DE。。。。。都不行。自己琢磨的解决思路是：1.输入ifconfig正常的是：而我的...
2018-09-28 21:42:06
7266
2
原创 ubuntu16.04下安装tensorflow
输入命令：pip install tensorflow-gpu==1.4.1ps:cuda:8.0cudnn:6.0cuda版本查看：cat /usr/local/cuda/version.txtcudnn 版本查看：cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2...
2018-09-20 16:58:48
97
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人