pytorch
文章平均质量分 52
qxq_sunshine
持之以恒
展开
-
模型的指数移动平均EMA
1、概念指数移动平均(Exponential Moving Average,EMA),也叫权重移动平均(Weighted Moving Average),是一种给予近期数据更高权重的平均方法。就是说,用原理数据来影响现在数据的更新。通俗版本理解:EMA是将每次梯度更新后的权值和前一次的权重进行联系,使得本次更新收到上次权值的影响。2、原理公式:a代表衰减率,该衰减率用于控制模型更新的速度,一般设为0.9-0.999。该值越大表示与上一次的影响越大,本次权重变化越小,与上次权重越接原创 2022-01-13 12:15:08 · 2873 阅读 · 0 评论 -
apex windows 安装步骤
第一步:直接上github上下载apex-master。链接为:GitHub - ptrblck/apex: A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch第二步:解压后打开,可以看到requirements.txt文件夹,里面写了依赖及版本要求,需要将里面写的依赖装好,具体操作为:进入anaconda的虚拟环境,然后进入倒apex的解压目录下,执行命令:python setup原创 2021-11-22 10:27:02 · 3043 阅读 · 2 评论 -
pytorch 版本信息造成的错误
1......torch/autograd/__init__.py", line 90, in backward allow_unreachable=True) # allow_unreachable flagRuntimeError: one of the variables needed for gradient computation has been modified by an i...翻译 2019-03-04 16:40:55 · 6571 阅读 · 0 评论 -
python 文件夹中所有图像旋转5度
1, 提供的代码分别为对文件夹中所有代码旋转5度,和对单张图像旋转5度的操作.缺点:有可能处理的图像不是原对应的图像.造成标签错乱. 更加科学的做法如2#coding=utf-8#对文件夹中的所有图片进行5度的旋转操作import PIL.Image as imgfrom PIL import ImageEnhanceimport osdef rotationImage(f...原创 2019-03-07 15:18:18 · 1275 阅读 · 0 评论 -
python 增加图像对比度
本代码实现的是,在旋转10度的基础上,再进行增加对比度的操作.1 代码:代码注释中的代码都是可以运行的. 但是不怎么靠谱,因为文件名被逐个编辑,有可能与原标签不对应,,更好的做法参考代码2# -*- coding: UTF-8 -*-from PIL import Imagefrom PIL import ImageEnhanceimport PIL.Image as img...原创 2019-03-07 18:00:54 · 3367 阅读 · 0 评论 -
报错:RuntimeError: CUDA error: all CUDA-capable devices are busy or unavailable
运行程序时报错:定位的错误是:D=torch.nn.DataParallel(discriminator(n_filters=32)).cuda(gpu_id)本来以为是因为服务器上有4块显卡,GPU1,GPU2,GPU3的计算能力都是2.0,但是GPU0的计算能力是>3的,所以认为是调用的时候为每个GPU都分配一定的任务,所以才报错的. 但是 看其他框架下的代码直接使用XX....原创 2019-03-11 19:33:16 · 14264 阅读 · 1 评论 -
精度输出遇到的问题
1. GAN网络输出的血管图不是二值图,而是RGB格式的图像;另外,输出的图像的大小跟输入的图像大小不一样,需要中心裁剪成原图像大小才能性能衡量.Question1:输入图像大小为565*584 ,但是输出的图像却是568*584 ??? 难道这个是因为网络的问题???2. 转化成灰度图后,输出的图像不是1或255,这一步需要转化!!! 但是别人的图片中的像素值也不是0或者255,...原创 2019-03-21 10:55:11 · 106 阅读 · 0 评论 -
Resnet特点总结
来自博客:https://www.jianshu.com/p/ca6bee9eb888设计初衷:残差结构是为了解决网络退化的问题提出的,梯度消失/爆炸已经通过 normalized initialization 等方式得到解决。1. 学习结果对网络权重的波动变化更加敏感。设 input :x , output :y , 待学习的 layer 参数为 w, 在 w 的改变增量一致时,...翻译 2019-05-19 14:57:57 · 18484 阅读 · 0 评论 -
安装指定版本的pytorch
1. pytorch0.4.0, python3.5, Linux, cu80, 86_64http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whl2. pytorch0.4.0, python3.5, Linux, cu90, 86_64http://download.pytorch.org...原创 2019-08-26 10:51:44 · 2158 阅读 · 0 评论 -
pytorch笔记 文件名的提取(后面,7)
1.模型保存与加载(1)保存和加载模型的参数:保存:torch.save(the_model.state_dict(),PATH)加载:the_model.load_state_dict(torch.load(PATH))(2)保存和加载整个模型torch.save(the_model,PATH)the_model=torch.load(PATH)2.每次做反向传播之前...翻译 2019-03-04 15:00:40 · 1882 阅读 · 0 评论 -
mkdir 和makedirs的区别
1. mkdir(path,[mode])作用:创建一个目录,可以是相对路径或者绝对路劲,模式是0777(任何用户都能访问)例如import osos.mkdir('./image') #成功#如果download文件夹存在,则成功#如果不存在就会报错 get a os errortry: os.mkdir('download/image')except OSError...翻译 2019-03-09 19:55:18 · 7805 阅读 · 0 评论 -
pytorch报错汇总
1、出现代码错误:module 'torchvision.transforms' has no attribute 'Resize'原因: torchvision版本较低解决方案:pip install --upgrade torchvision 升级torchvision到 0.2.1 版本。 (问题解决)...原创 2018-12-13 16:47:25 · 682 阅读 · 0 评论 -
转置卷积(反卷积)
借鉴于此个博客,写的非常好:https://blog.csdn.net/isMarvellous/article/details/80087705转置卷积(Transposed Convolution)也叫做反卷积(Deconvolution)Pytorch中可以使用torch.nn.ConvTranspose2d()来调用,caffe中对应的转置卷积层为deconv_layer作用:...翻译 2018-12-28 14:29:51 · 425 阅读 · 0 评论 -
图像的超分辨率重建SRGAN与ESRGAN
SRGAN传统的图像超分辨率重建方法一般都是放大较小的倍数,当放大倍数在4倍以上时就会出现过度平滑的现象,使得图像出现一些非真实感。SRGAN借助于GAN的网络架构生成图像中的细节。训练网络使用均方误差(MSE)能够获得较高的峰值信噪比(PSNR),但是恢复出来的图像会丢失图像的高频细节信息,在视觉上有不好的体验感。SRGAN利用感知损失(perceptual loss)和对抗损失(adv...原创 2018-12-23 20:59:59 · 16335 阅读 · 6 评论 -
pytorch一些常用的操作 总结
https://blog.csdn.net/TH_NUM/article/details/830889151、tensor的拼接#coding=utf-8import torch#张量的拼接x=torch.randn(2,3)print(x)x1=torch.cat((x,x,x),0) #得到tensor大小:(6, 3)print("x1",x1,"x1.size()"...翻译 2019-01-11 22:37:58 · 1318 阅读 · 0 评论 -
pytorch的padding的理解和操作
1 padding 的操作就是在图像块的周围加上格子, 从而使得图像经过卷积过后大小不会变化,这种操作是使得图像的边缘数据也能被利用到,这样才能更好地扩张整张图像的边缘特征.公式表示如下:2 卷积核中size的选择可能会导致input中的某几行(或者最后 几行)没有关联起来,这个可能是因为我们使用的模式是valid,而不是full(tensorflow中也叫做same模式,不过这...翻译 2019-01-13 22:35:47 · 35347 阅读 · 5 评论 -
pytorch 问题
1 遇到问题报错:RuntimeError: cudaEventSynchronize in future::wait: device-side assert triggered一个博客的作者解释这个问题如下:这个问题很诡异。。如果你用的是BCECriterion(),然后你的网络最后一层不是sigmoid,就会出这个错误。因为BCECriterion的期望输入是[0,1]。不加sig...原创 2019-01-15 10:33:25 · 1093 阅读 · 0 评论 -
python中的图像旋转和仿射变换
python图像的读取,保存,复制操作:https://blog.csdn.net/qq_40962368/article/details/80444144python中实现图像旋转的代码如下(亲测,可执行)#coding=utf-8import cv2import numpy as npimg=cv2.imread('cat.jpg')print(img.shape)row...翻译 2019-03-05 10:02:11 · 896 阅读 · 0 评论 -
python 图像的旋转+翻转操作+PIL读取图像 +对比度,锐度,色度,亮度增强
#coding=utf-8import cv2import numpy as npfrom numpy import *#**************旋转操作***************img=cv2.imread('21_manual1.png')rows,cols,channel=img.shapefor j in range(0, 71): M=cv2.getRot...原创 2019-03-05 22:10:40 · 6357 阅读 · 3 评论 -
图像的翻转操作
具体代码执行如下:#coding=utf-8from PIL import Imageimport osimport os.path#图片的水平翻转rootdir = r'/home/qxq/Desktop/eyedata_final/mask/rotation/' # 指明被遍历的文件夹for parent, dirnames, filenames in os.walk(ro...原创 2019-03-08 15:06:36 · 2237 阅读 · 0 评论 -
pytorch错误
运行代码报错:invalid gradient at index 0 - expected shape [] but got [1]定位的代码段是: d_real.backward(mone)其中mone的定义如下:one=torch.Tensor([1])mone=one*-1mone=mone.cuda()不知道是为什么,但是后来将 对应的代码段改为d_real.ba...原创 2019-03-09 15:46:56 · 2050 阅读 · 0 评论 -
pytorch载入和显示图片
代码实现如下:import numpy as npfrom torchvision import transformsimport cv2#transforms.ToTensor()transform1=transforms.Compose([transforms.ToTensor()])#读入图像img=cv2.imread('cat.jpg')img1=transform...原创 2018-12-07 21:44:54 · 960 阅读 · 0 评论