pytorch
Shuai@
这个作者很懒,什么都没留下…
展开
-
pytorch模型参数量和模型推理速度
PyTorch模型参数量和推理速度计算原创 2022-12-02 20:33:38 · 1317 阅读 · 0 评论 -
nnUNet ubuntu环境配置用2D图像训练
nnUnet原创 2022-07-26 14:59:59 · 1208 阅读 · 2 评论 -
动态卷积的本质
动态卷积的计算量增加很少原创 2022-07-19 00:41:06 · 215 阅读 · 0 评论 -
使用深度学习训练图像时,图像太大进行切块训练预测
将图像进行切块原创 2022-07-14 17:22:35 · 1146 阅读 · 0 评论 -
PyTorch 使用 MaxPool 实现图像的膨胀和腐蚀
PyTorch 使用 MaxPool 实现图像的膨胀和腐蚀转载 2022-07-12 16:57:44 · 538 阅读 · 0 评论 -
深度学习医学图像模型复现
经典方法复现原创 2022-06-28 15:12:34 · 374 阅读 · 0 评论 -
pytorch 计算模型的GFlops和total params的方法
pytorch计算参数和Gflops原创 2022-06-16 16:40:37 · 2797 阅读 · 0 评论 -
深度学习热力图可视化的方式
热力图可视化原创 2022-06-11 09:50:31 · 1267 阅读 · 0 评论 -
pytorch使用hook获得特征图
pytorch 可视化特征原创 2022-06-09 19:39:59 · 650 阅读 · 0 评论 -
mmsegmentation框架配置
首先创建一个虚拟环境conda create -n open-mmlab python=3.7 -yconda activate open-mmlab可以参考mmsegmention的官网安装文档首先安装pytorch框架(GPU)pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.htmlpip ins原创 2022-03-08 20:17:56 · 1496 阅读 · 0 评论 -
pytorch 中的张量如何修改数据类型
tensor张量有不同的数据类型,例如torch.long,torch.float32,torch.float16等之间可以相互转化import torchx = torch.zeros([2,2])x = x.to(torch.long)x = x.to (torch.float32)原创 2021-10-19 19:48:04 · 2687 阅读 · 0 评论 -
pytorch 查看优化optimizer其中参数的学习率
from torchvision.models import resnet18import torchmodel = resnet18()param = torch.zeros([2,2])optimizer = torch.optim.Adam([param],lr=1e-4)for param_group in optimizer.param_groups:# print(param_group) output:{'params': [tensor([[0., 0.],原创 2021-10-19 19:44:42 · 2902 阅读 · 0 评论 -
ValueError: only one element tensors can be converted to Python scalar
ValueError: only one element tensors can be converted to Python scalar使用torch.tensor 将list列表转化成pytorch中的tensor张量时,如果遇到如下报错,很有可能是因为list中包含多个list数据,或者维度不相同的数据。使用torch.stack 将list中数据转化成tensor。torch.tensor([a,b])#换成stacktorch.stack([a,b])...原创 2021-08-16 14:09:15 · 632 阅读 · 0 评论 -
torch.cat 和 torch.stack
torch.cat 和 torch.stack看起来相似但是性质还是不同的使用python中的list列表收录tensor时,然后将list列表转化成tensor时,会报错。这个时候就要使用torch.stack进行堆叠,转化成tensor。torch.cat()torch.cat(tensors,dim=0,out=None)→ Tensortorch.cat()对tensors沿指定维度拼接,但返回的Tensor的维数不会变import torcha = torch.rand((2, 3原创 2021-08-12 14:55:06 · 951 阅读 · 0 评论 -
使用pytorch的dataloader来平衡不平衡数据的抽取
参考聊聊Pytorch中的dataloader定义sample:(假设dataset类返回的是:data, label)from torch.utils.data.sampler import WeightedRandomSampler## 如果label为1,那么对应的该类别被取出来的概率是另外一个类别的2倍weights = [2 if label == 1 else 1 for data, label in dataset]sampler = WeightedRandomSampler(转载 2021-07-05 16:49:38 · 5052 阅读 · 0 评论 -
2021-07-01
Anaconda 镜像使用帮助参考清华官方anaconda 镜像源官网说明,网站链接anacondaAnaconda 是一个用于科学计算的 Python 发行版,支持 Linux, Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包。Anaconda 安装包可以到 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 下载。TUNA 还提供了 Anaconda 仓库与第三方源(conda-forge、msys2、转载 2021-07-01 09:56:37 · 65 阅读 · 0 评论 -
修正的轻量型UNet网络
import torch.nn as nnimport torchfrom tensorboardX import SummaryWriter'''2021/6/1problem1.nn.Batch2.Conv2dtranpose-nn.ConvTranspose2d(in_channels=out_c,out_channels=out_c,kernel_size=4,stride=2)3.dropout(p)4.deblock (2*in_c,in_c,out_c)??'''..原创 2021-06-01 15:52:36 · 586 阅读 · 0 评论 -
PyTorch图像处理器torchvision.ToTensor-归一化
ToTensor使用时一定要小心呀它会把图像从[0-255]变到[0-1]之间如果想把图像由numpy 转成 tensor的话,可以使用如下命令img = torch.tensor(img)原创 2020-12-11 11:28:04 · 2052 阅读 · 0 评论 -
pytorch 使用GPU计算f1_score
pytorch使用GPU计算评价指标如下是参考链接f1score with GPUdef f1_loss(y_true:torch.Tensor, y_pred:torch.Tensor, is_training=False) -> torch.Tensor: '''Calculate F1 score. Can work with gpu tensors The original implmentation is written by Michal Haltuf on转载 2020-12-09 18:56:38 · 5391 阅读 · 3 评论 -
pytorch使用dataloader进行数据数据加载
1.建立一个图片路径的列表及List,每一个元素存放input和output。2.然后讲这样的列表送入data类里面。进行数据读取,和数据增强然后反回img,label。3.然后将data类和dataloader结合,dataloader能够触发data类中的__getiterm__函数,不停地反回batch数据。...原创 2020-12-08 10:18:46 · 714 阅读 · 0 评论 -
visdom首次开启下载耗时+visdom蓝屏处理办法
以上问题都还是visdom首次开启需要下载一些文件,但是地址是国外的,所以下载慢或者下载不下来。主要是这个文件环境中visdom/static不全。如下链接是完整的static文件,替换环境中的static就行了。链接:https://pan.baidu.com/s/1wYr7d1ZRj7x-fWZbbAF4Ag提取码:wjl5...原创 2020-11-19 17:36:26 · 639 阅读 · 1 评论 -
pytorch参数固定
for i in m.parameters(): i.requires_grad=False固定之后也要使用filter训练参数时只训练requires_grad为Ture的参数。optimizer.SGD(filter(lambda p: p.requires_grad, model.parameters()), lr=1e-3)原创 2020-11-19 15:29:07 · 746 阅读 · 0 评论 -
pytorch网络图可视化
tensorboardX可视化pytorch网络图首先python 环境中要安装一下的库pip install tensorboardpip install tensorflowpip install tensorboardX运行如下demo代码来源于:https://blog.csdn.net/sunqiande88/article/details/80155925import torchimport torch.nn as nnfrom tensorboardX import.原创 2020-11-12 10:55:37 · 322 阅读 · 1 评论