解决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...
Pytorch 抽取vgg各层并进行定制化处理的方法
yipeiwu_com6年前
工作中有时候需要对vgg进行定制化处理，比如有些时候需要借助于vgg的层结构，但是需要使用的是2 channels输入，等等需求，这时候可以使用vgg的原始结构用class重写一遍，但是...
pytorch 预训练层的使用方法
yipeiwu_com6年前
pytorch 预训练层的使用方法 将其他地方训练好的网络，用到新的网络里面 加载预训练网络 1.原先已经训练好一个网络 AutoEncoder_FC() 2.首先加载该网络，读取其存储...
