python
一杯盐水
这个作者很懒,什么都没留下…
展开
-
python关于onnx模型的一些基本操作
onnx的基本操作一、onnx的配置环境二、获取onnx模型的输出层三、获取中节点输出数据四、onnx前向InferenceSession的使用1. 创建实例,源码分析2. 模型推理run,源码分析五、遇到的一些问题最近在对模型进行量化时候,模型格式转变为onnx模型了,因此需要对onnx进行加载、运行以及量化(权重/输入/输出)。故,对onnx模型的相关操作进行简单的学习,并写下了这边博客,若有错误请指出,谢谢。一、onnx的配置环境onnx的环境主要包含两个包onnx和onnxruntime,我们原创 2020-09-23 18:16:09 · 33776 阅读 · 6 评论 -
KNN实现MNIST数字分类
KNN实现MNIST数字分类import torchfrom torch.utils.data import DataLoaderimport torchvision.datasets as dsetsimport numpy as npimport operatorbatch_size = 100data_path = './'def KNN_distance(k, dis, trains_1, labels_2, test): #曼哈顿距离(Manhattan distan原创 2020-06-22 22:55:09 · 985 阅读 · 1 评论 -
深度学习之——KNN算法(k-最近邻算法)
k-最近邻算法基础理论欧式距离(Euclidean distance)曼哈顿距离(Manhattan distance)代码实例 在学习深度和图像识别的时候,看见了一个比较有意思的算法——KNN算法,该算法是图像分类中最简单的算法之一。基础理论 KNN算法全称是K-最近邻算法,英文名称是K-NearestNeighbor,简称为KNN;从算法名称上,可以猜出,是找到最近的k个邻居,在选取到的k个样本中选取出最近的且占比最高的类别作为预测类别。如下图所示:KNN算法 上图所指示,蓝色的正方形原创 2020-05-15 18:46:03 · 7185 阅读 · 0 评论 -
详解激活函数Mish
激活函数Mish 对YOLO有了解的或者从事人工智能的相关人员,都知道前段时间yolov4出来了,而且相对于yolov3在精度上有一个质的飞跃,让人感觉匪夷所思,于是在好奇心的驱使下,了解了一下yolov4;与yolov3相比较,主要有几个不同点:网络层更深了、激活函数将leakyRelu替换为mish、采用了SPP(空间金字塔池化)等。下面主要学习一下激活函数mish: 在论文“Mish: A Self Regularized Non-Monotonic Neural Activation Fun原创 2020-05-13 16:16:50 · 12271 阅读 · 3 评论 -
神经网络CNN之池化层
池化层简述池化层的分类最大/均值池化中值池化组合池化Spatial Pyramid PoolingGlobal Average/Max Pooling参考文献简述 池化层(Pooling Layer)是CNN中常见的一种操作,池化层通常也叫做子采样(subsampling)或降采样(Downsampling),在构建CNN网络时,往往是用在卷积层之后,通过池化层来降低卷积层输出的特征维度,在有效减少网络参数的同时还可以防止过拟合现象。 说到池化操作,就会想到我们经常用的池化操作,即最大池化(Max原创 2020-05-08 18:06:16 · 6496 阅读 · 0 评论 -
pytorch保存onnx模型
pytorch保存onnx模型因为一些原因,需要用pytorch去创建、训练和保存模型。pytorch保存的模型通常为pth、pt、pkl的格式,但这种类型的模型不能在其他框架(tensorflow)下直接加载,因此需要将模型保存为其他格式的。在网上进行相应的学习后,总结出一下两点:pytorch可以直接将模型保存为onnx的,并且可以通过onnx转换为其他格式的模型(pb);pytorc...原创 2020-04-07 11:20:57 · 5516 阅读 · 3 评论 -
pytorch加载.pth格式文件
pytorch怎么读取.pth文件,以及怎么将torch的.pth文件转化为bin文件进行保存。import torchimport numpy as nppath_in="in.pth"path_out="out.pth"#读取文件#若安装的pytorch是带Cuda的版本,cuda不能用,则需要指定为cpu运行data_in = torch.load(path_in, 'cp...原创 2020-03-30 15:43:27 · 2582 阅读 · 0 评论 -
Linux安装pytorch及查看版本
目录Linux安装pytorch安装命令验证事例查看pytorch版本Linux安装pytorch安装命令# Python 2.7pip install https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp27-cp27mu-linux_x86_64.whlpip install torchvision# if the ab...原创 2020-03-30 13:50:20 · 4784 阅读 · 0 评论