使用卷积对图片进行处理 利用tensorboard进行查看卷积后的图片
通常只使用这五个参数,
import torch
from torch import nn
import torch.nn.functional as F
import torchvision
from torch.nn import Conv2d
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
dataset = torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor(),download=True)
dataloader = DataLoader(dataset,batch_size=64)
class juanji(nn.Module):
def __init__(self):
super(juanji, self).__init__
共有 0 条评论