Pytorch学习笔记之卷积层处理图片并进行可视化

1793-席同学

发表文章数:14

首页 » Python » 正文

实战卷积层

import torch
from torch import nn
from torch.nn import Conv2d
from torch.utils.data import DataLoader
import torchvision
from tensorboardX import SummaryWriter

dataset = torchvision.datasets.CIFAR10('./data',
                                       train=False,
                                       transform=torchvision.transforms.ToTensor(),
                                       download=True)
dataloader = DataLoader(dataset, batch_size=64)


class qingfeng(nn.Module):
    def __init__(self):
        super(qingfeng, self).__init__()
        self.conv1 = Conv2d(in_channels=3,
                            out_channels=6,
                            kernel_size=3,
                            stride=1,
                            padding=0)

    def forward(self, x):
        x = self.conv1(x)
        return x


qingfeng = qingfeng()
print(qingfeng)
writer = SummaryWriter('./logs')
step = 0
for data in dataloader:
    imgs, targets = data
    print("输入图片形状",imgs.shape)
    output = qingfeng(imgs)
    print("输出图片形状",output.shape)
    writer.add_images("input", imgs, step)

    output = torch.reshape(output, (-1, 3, 30, 30))
    print("改变形状后的图像",output.shape)
    writer.add_images('output', output, step)
    step += 1
writer.close()

activate pytorch
tensorboard –logdir “logs”


未经允许不得转载:作者:1793-席同学, 转载或复制请以 超链接形式 并注明出处 拜师资源博客
原文地址:《Pytorch学习笔记之卷积层处理图片并进行可视化》 发布于2021-10-17

分享到:
赞(0) 打赏

评论 抢沙发

评论前必须登录!

  注册



长按图片转发给朋友

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

Vieu3.3主题
专业打造轻量级个人企业风格博客主题!专注于前端开发,全站响应式布局自适应模板。

登录

忘记密码 ?

您也可以使用第三方帐号快捷登录

Q Q 登 录
微 博 登 录