2021年12月_北京纯牛奶

转载 PyTorch 普通卷积和空洞卷积

import numpy as npfrom torchvision.transforms import Compose, ToTensorfrom torch import nnimport torch.nn.init as initdef transform(): return Compose([ ToTensor(), # Normalize((12,12,12),std = (1,1,1)), ])arr = range(1,26)arr

2021-12-14 20:55:49 389

原创 pytorch可视化输出模型计算力和参数量，设备损耗，运行时间等信息

输出下面的表格形式的，将自己的模型都输出出来。from torchstat import statfrom torchvision.models import resnet50model = resnet50()stat(model, (3, 224, 224)) # (3,224,224)表示输入图片的尺寸记得：pip install torchstat不过好像会报下面的错误：所以，只需要在最下面的错，点进去了。将detach()后面变为detach().cpu,bu，不过还是会报

2021-12-10 16:20:17 1769

原创 pytorch调用不同的分类模型

其实pytorch内部已经嵌入了需要的分类模型，包括：resnet、densenet、AlexNet、vgg、SqueezeNet、SqueezeNet、googlenet等等只需要自己直接调用即可调用方法：import torchvisionmodel = torchvision.models.resnet50(pretrained=True)不需要预训练模型的参数来初始化model = torchvision.models.resnet50(pretrained=False)pretra

2021-12-10 13:09:27 1719

原创 pytorch自己搭建深度学习图像分类模型

首先导入一些必须使用的包，这些包导入的包都是后续操作用的，包括pytorch对数据集加载等一些操作；看你自己需要使用什么功能，就导入什么样子的包，不过大致需要导入的包是相同的import torchfrom torch import nnfrom net import MyAlexNetimport numpy as npfrom torch.optim import lr_schedulerimport osfrom torchvision import transformsfrom .

2021-12-08 11:28:31 1238

原创对图片数据集进行划分代码

以下代码来源：https://www.bilibili.com/video/BV18L4y167jr?p=3&spm_id_from=pageDriver这个博主，首先感谢他制作的视频教程，个人觉得写的还不错其源代码为：import osfrom shutil import copyimport randomdef mkfile(file): if not os.path.exists(file): os.makedirs(file)# 获取data文件

2021-12-08 11:01:21 910

原创查看显卡和cuda是否安装成功

我被这个问题尴尬了很久，nvidia-smi 查看显卡驱动是否安装成功nvcc -V 查看cuda是否安装成功，其实也就是cuda是否安装成功记住这两个都需要网上下载对应版本的软件，下载之后正常跟安装其他的软件一样安装就可以，但是cuda安装的时候，一会也会涉及安装pytorch,别去最新官网安装了，因为你已经安装了cuda了，所以找对应版本的安装就可以，网址：https://pytorch.org/get-started/previous-versions/

2021-12-05 21:08:32 4013

原创本机cuda和环境cuda版本不对应导致没办法调用

有的的时候，自己新建的环境和本地安装的cuda版本不对应，会导致自己的没办法使用GPU运行nvcc -V 显示的是本机的cuda版本，torch.version.cuda 显示的是该环境的版本，记得在import torch 里面进行输入因为两个版本需要对应，最好就是本机安装什么cuda版本，环境就安装什么cuda版本，目前我是这样解决的...

2021-12-05 21:00:30 1928 4

原创安装显卡驱动，每次关机都掉显卡驱动

下载官网网址：https://www.nvidia.cn/drivers/results/181582/安装：sudo chmod a+x NVIDIA-Linux-x86_64-455.38.run sudo ./NVIDIA-Linux-x86_64-455.38.run之后一直确定就可以

2021-12-03 21:40:22 2455

原创全部图片数据集划分8：2数据集，python代码

划分数据集是指全部的数据集，之后划分成train占8份，test占两份，划分之后是每一类下面都有一个train和test文件首先，新建一个空文件，子目录下把每一类的文件夹复制，新建代码pythonimport os# i 用来计算文件数量，k 用来计算新建文件夹的数量i = 0k = 0# 想保存到的根路径save_dir = r'C:\Users\Administrator\Desktop\project\256_ObjectCategories_New5'# 如果目录不存在，则创

2021-12-01 12:00:26 2828

微信号：Custom-Software