使用 tensorboard 展示输入图像及最大池化后的图像,代码如下所示。
from torch import nn
from torch.nn import MaxPool2d
import torchvision.datasets
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
test_data = torchvision.datasets.CIFAR10("./dataset", train=False,
transform=torchvision.transforms.ToTensor())
test_loader = DataLoader(dataset=test_data, batch_size=4)
class Tudui(nn.Module):
def __init__(self):
super(Tudui, self).__init__()
self.maxpool1 = MaxPool2d(kernel_size=3, ceil_mode=False)
def forward(self, input):
output = self.maxpool1(input)
return output
tudui = Tudui() # 创建神经网络
writer = SummaryWriter("logs_maxpool")
step = 0
for data in test_loader:
imgs, targets = data
writer.add_images("input", imgs, step)
output = tudui(imgs)
writer.add_images("ouput", output, step)
step += 1
writer.close()
运行上述代码。
运行结束后,打开 Pycharm 中的 Terminal ,执行如下命令行并得到相应网址。
随后打开网址,得到如下。