import torch.nn as nn
from torchvision import transforms
from PIL import Image
import torch
img = Image.open('../data/pikachu.jpg')
transform_to_tensor = transforms.ToTensor()
img_ = transform_to_tensor(img)
print(1, img_.size())
img_ = torch.unsqueeze(img_, 0)
conv = nn.Conv2d(3, 3, kernel_size=7, padding=3)
img_ = conv(img_)
img_ = conv(img_)
img_ = conv(img_)
img_ = conv(img_)
img_ = torch.squeeze(img_, 0)
print(2, img_.size())
unloader = transforms.ToPILImage()
out = unloader(img_)
out.show()