pytorch 使用单个GPU与多个GPU进行训练与测试的方法
yipeiwu_com6年前
如下所示： device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")#第一行代码 model.t...
将Pytorch模型从CPU转换成GPU的实现方法
yipeiwu_com6年前
最近将Pytorch程序迁移到GPU上去的一些工作和思考 环境：Ubuntu 16.04.3 Python版本：3.5.2 Pytorch版本：0.4.0 0. 序言 大家知道，在深度学...
解决pytorch GPU 计算过程中出现内存耗尽的问题
yipeiwu_com6年前
Pytorch GPU运算过程中会出现：“cuda runtime error(2): out of memory”这样的错误。通常，这种错误是由于在循环中使用全局变量当做累加器，且累加...
python opencv实现证件照换底功能
yipeiwu_com6年前
本文实例为大家分享了python opencv实现证件照换底功能的具体代码，供大家参考，具体内容如下 思路：先转到HSV空间，利用颜色提取背景制作掩模版mask，然后通过按位操作提取人像...
用Pytorch训练CNN(数据集MNIST,使用GPU的方法)
yipeiwu_com6年前
听说pytorch使用比TensorFlow简单，加之pytorch现已支持windows，所以今天装了pytorch玩玩，第一件事还是写了个简单的CNN在MNIST上实验，初步体验的确...
pytorch多进程加速及代码优化方法
yipeiwu_com6年前
目标：优化代码，利用多进程，进行近实时预处理、网络预测及后处理： 本人尝试了pytorch的multiprocessing，进行多进程同步处理以上任务。 from torch.mul...
python实现证件照换底功能
yipeiwu_com6年前
本来是在找交通识别的程序，然后凑巧看见了证件照换底，于是学习了一下~一开始在网上找了一个很普遍写的程序，但是效果并不好，想要放弃了，然后看见了这个，参考：python opencv实现证...
pytorch在fintune时将sequential中的层输出方法,以vgg为例
yipeiwu_com6年前
有时候我们在fintune时发现pytorch把许多层都集合在一个sequential里，但是我们希望能把中间层的结果引出来做下一步操作，于是我自己琢磨了一个方法，以vgg为例，有点僵硬...
pytorch 在sequential中使用view来reshape的例子
yipeiwu_com6年前
pytorch中view是tensor方法，然而在sequential中包装的是nn.module的子类， 因此需要自己定义一个方法： import torch.nn as nn c...
python实现抠图给证件照换背景源码
yipeiwu_com6年前
本文实例为大家分享了python实现抠图给证件照换背景的具体代码，供大家参考，具体内容如下 import cv2 import numpy as np import matplotl...
