PyTorch实现卷积和池化（kaggle环境）

import numpy as np
import torch
from torch import nn
from torch.autograd import Variable
from PIL import Image
import matplotlib.pylab as plt
%matplotlib inline

卷积层

导入图片并且转化为灰度图

kaggle环境中所有本地数据都有上传才能加载！

im = Image.open('../input/cat.jpg').convert('L')

将图片转化为矩阵

im = np.array(im, dtype='float32')

查看图片，将图片矩阵转化为uint8类型，uint8是无符号八位整型，表示范围是[0, 255]的整数，colormap为gray

plt.imshow(im.astype('uint8'), cmap='gray')

<matplotlib.image.AxesImage at 0x7f301f697160>

im.shape