![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
代码小白的成长
不管多么难熬,人生都要眉开眼笑
展开
-
特征图的显示
【代码】特征图的显示。原创 2023-02-27 12:21:46 · 223 阅读 · 0 评论 -
输出网络参数量
【代码】输出网络参数量。原创 2023-02-16 20:13:30 · 127 阅读 · 0 评论 -
根据特征图绘制热图
【代码】根据特征图绘制热图。原创 2023-02-08 12:18:54 · 509 阅读 · 0 评论 -
NVIDIA GeForce RTX 3090 Ti with CUDA capability sm_86 is not compatible with the current PyTorch
RTX 3090的算力是8.6,但是当前的PyTorch依赖的CUDA版本支持的算力只有3.7、5.0、6.0、7.0算力7.0的显卡可以在支持最高算力7.5的CUDA版本下运行,但是算力7.5的显卡不可以在支持最高算力7.0的CUDA版本下运行。同理算力8.x的显卡不可以在支持最高算力7.x的CUDA版本下运行解决办法:重新安装cuda和cudann。原创 2022-09-18 18:53:27 · 2215 阅读 · 0 评论 -
Slurm超算集群跑深度学习代码教程
参考教程http://docs.hpc.whu.edu.cn/如果是校外操作,则需要通过easyconnect登陆集群专属VPNhttps://vpn.whu.edu.cn/1.连接超算使用支持 SSH 协议的客户端软件MobaXterm来登录超算https://mobaxterm.mobatek.net/使用 SSH 客户端连接超算的登录节点 swarm.whu.edu.cn用户存储与数据传输文件传输(SFTP 协议)服务器IP地址:202.114.96.177anaconda原创 2022-06-01 15:21:47 · 5060 阅读 · 0 评论 -
将原始的Helen数据集生成彩色的人脸分割图像
Helen人脸数据集包括2330张人脸图像,并且每张人脸图像包含11个部位的掩模,通常做人脸分割只需要面部以及面部特征掩模即可。数据集文件的整体结构如下:原创 2021-05-13 10:32:16 · 612 阅读 · 4 评论 -
VGG下载(.net文件和imagenet-vgg-verydeep-19)
1、imagenet-vgg-verydeep-19链接:https://pan.baidu.com/s/1PU9cVExFXb0afXnTEHYg6Q 提取码:2okb2、vgg19.npy链接:https://pan.baidu.com/s/1Rb5fZ3vBNzd4YPbo93ys4w提取码:9pr03、vgg16.npy链接:https://pan.baidu.com/s/1D8469KgIBitWs-9QT834vQ提取码:9zmh...原创 2020-09-20 15:43:35 · 3980 阅读 · 4 评论 -
创建好后的模型,对Con2d, ConvTranspose2d ,以及归一化BatchNorm2d函数中的变量进行初始化
def weights_init(m): classname = m.__class__.__name__ if (classname.find('Conv2d') != -1) | (classname.find('ConvTranspose2d') != -1): m.weight.data.normal_(0.0, 0.02) elif classname.find('BatchNorm2d') != -1: m.weight.dat.原创 2020-09-02 17:50:31 · 560 阅读 · 1 评论 -
图像质量评价指标: MMD ( maximum-mean-discrepancy) 最大平均差异
MMD:maximum mean discrepancy。最大平均差异, 用于判断两个分布p和q是否相同。它的基本假设是:如果对于所有以分布生成的样本空间为输入的函数f,如果两个分布生成的足够多的样本在f上的对应的像的均值都相等,那么那么可以认为这两个分布是同一个分布。现在一般用于度量两个分布之间的相似性。import torchimport matplotlibimport osimport argparseimport numpy as npfrom PIL import Imagefr原创 2020-08-31 23:03:53 · 2625 阅读 · 1 评论 -
readImg: 读取图片到Variable变量
from PIL import Imagefrom torch.autograd import Variablefrom torchvision import transformstransform = transforms.Compose( [ # 函数接受PIL Image或numpy.ndarray,将其先由HWC转置为CHW格式,再转为float后每个像素除以255 transforms.ToTensor(), # 对数据按通道进行标准化到[-1,1原创 2020-08-31 20:49:58 · 1882 阅读 · 0 评论 -
图像评价指标 : PSNR (Peak Signal-to-Noise Ratio) 峰值信噪比 和 SSIM (Structural SIMilarity) 结构相似性
# example of calculating the frechet inception distance in Kerasimport osimport torchimport argparseimport numpy as npfrom PIL import Imagefrom skimage.measure import compare_psnr, compare_ssimdef dataloader(dataset): dataloader = torch.utils.原创 2020-08-28 21:07:09 · 619 阅读 · 0 评论 -
图片保存:torchvision.utils.save_image(img, imgPath)
torchvision.utils.save_image(img, imgPath)深度学习模型中,一般使用如下方式进行图像保存(torchvision.utils中的save_image()函数),这种方式只能保存RGB彩色图像,如果网络的输出是单通道灰度图像,则该函数依然会输出三个通道,每个通道的数值都是相同的,即“伪灰度图像”,虽然从视觉效果上看不出区别,但是图像所占内存比正常情况大了两倍。save_image()函数:def save_image(tensor, filename, nro原创 2020-08-24 13:34:36 · 24203 阅读 · 1 评论