Infi-chu: http://www.cnblogs.com/Infi-chu/ torch.FloatTensor:用于生成数据类型为浮点型的Tensor，参数可以是一个列表，也可以是一个维度。 import to…

pytorch使用说明 1.torch和numpy的转换 import torch import numpy as np np_data = np.arange(6).reshape((2, 3)) torch_data…

PyTorch如今发布到1.1稳定版本，新增的功能让模型部署变得更为地简单，本文记录如何利用C++来调用PyTorch训练好的模型，其实也是利用官方强大的LibTorch库。 LibTorch的安装 虽然说安装，其实就是…

本demo从pytorch官方的迁移学习示例修改而来，增加了以下功能： 根据AUC来迭代最优参数； 五折交叉验证； 输出验证集错误分类图片； 输出分类报告并保存AUC结果图片。 1 import os 2 import …

其他：windows使用nvidia-smi查看gpu信息 为什么将数据转移至GPU的方法叫做.cuda而不是.gpu，就像将数据转移至CPU调用的方法是.cpu？这是因为GPU的编程接口采用CUDA，而目前并不是所有的…

这里简单介绍一下用PyTorch在CPU上的一些性能相关的BKM。 内容以inference为主，毕竟CPU上主要的场景还是inference；另外这里CPU都指的是Intel Xeon. gist里面写了英文版的，内容…

https://discuss.pytorch.org/t/how-the-pytorch-freeze-network-in-some-layers-only-the-rest-of-the-training/7088…

　　ConvTransposed2d()其实是Conv2d()的逆过程，其参数是一样的 　　Conv2d(): 　　　　output = (input+2*Padding-kernelSize) / stride + 1…

—恢复内容开始— MAP是衡量object dectection算法的重要criteria，然而一直没有仔细阅读相关代码，今天就好好看一下： 1. 测试test过程是由FRCN/tools/tes…

之前在用预训练的ResNet的模型进行迁移训练时，是固定除最后一层的前面层权重，然后把全连接层输出改为自己需要的数目，进行最后一层的训练，那么现在假如想要只是把 最后一层的输出改一下，不需要加载前面层的权重，方法如下： …

一、简化前馈网络LeNet 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 import …

yolo系列是目标识别的重头戏，为了更好的理解掌握它，我们必须从源码出发深刻理解代码。下面我们来讲解pytorch实现的yolov3源码。在讲解之前，大家应该具备相应的原理知识yolov1，yolov2，yolov3。 …