- 博客(44)
- 收藏
- 关注
原创 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
1197
原创 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
744
原创 Homography单应性矩阵
比如 上图中的 桌面,地面,橱柜面。对于这种图像中有多个平面的情况,我们就需要针对每一个平面使用其对应的Homography了。对于所有的对应点,只要它们都位于同一个物理平面上,上述 Homography 就是成立的。换句话说,就是可以把图一中书皮上的所有点都映射到图二的书皮上,也就是看起来,图一中的书皮和图二中的书皮对齐了!考虑 同一个平面(比如书皮)的两张图片,红点表示同一个物理坐标点在两张图片上的各自位置。将矩阵带入变换公式,可以得到实际的变换为:x' = w-x, y' = y。
2023-07-22 21:19:27
178
原创 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
528
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
1935
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
429
原创 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
1221
原创 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
1912
原创 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
849
原创 解决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
25640
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
365
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
4024
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
14866
原创 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
3250
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
4355
原创 安装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
3930
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
265
原创 ubuntu下查看电脑硬件的基本信息
1.查看cpu信息cat /proc/cpuinfo2.查看内存cat /proc/meminfo3.查看硬盘sudo fdisk -l | grep "Disk /dev/sd"
2020-05-30 11:18:48
2519
原创 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
2568
原创 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
6062
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
2081
原创 /boot空间不足的解决办法
1.查看系统已经安装的内核版本dpkg --get-selections |grep linux-image2.查看系统当前使用的内核版本uname -a3.常规利用删除旧版本内核的方法来解决该问题,但是总会遇到如下问题:有未能满足的依赖关系。请尝试不指明软件包的名字来运行“apt-get -f install”(也可以指定一个解决办法)。于是执行命令apt-...
2019-11-04 18:15:45
476
原创 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
388
原创 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
12893
原创 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
2883
原创 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
243
原创 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
952
原创 pytorch运行程序过程中遇到的问题
将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告,它在提醒用户下个版本这将成为一个错误错误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
1668
原创 安装驱动时的问题
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
6565
原创 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
2857
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
1473
原创 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
245
原创 终止正在运行的程序
Ubuntu下运行程序,如果按快捷键ctrl+c无法终止程序,可以按照如下步骤终止程序:nvidia-smi假如我们想关掉正在运行的caffe程序,看一下它的PID编号是21457,则执行以下命令即可终止程序kill -9 21457 ...
2019-01-19 10:41:14
6292
原创 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
2562
原创 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
1055
原创 降级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
3959
原创 /boot 100%
wanglin@wanglin-System-Product-Name:~/桌面$ df发现 /boot 100%出现这个问题的原因是系统的冗余内核太多。此时在终端下输入 再输入uname -a显示当前正在使用的系统内核发现正在使用的系统内核为4.15.0-36,因此需要卸掉冗余的内核。如果内核的状态是install,说明:系统已经安装了相应的内核,使用purge命令删除...
2018-11-13 21:33:36
629
原创 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
14055
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
7295
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
101
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人