使用卷积提取图片轮廓

一 实例描述

将彩色的图片生成带边缘化信息的图片。

本例中先载入一个图片，然后使用一个“3通道输入，1通道输出的3*3卷积核”（即sobel算子），最后使用卷积函数输出生成的结果。

二 代码

'''
载入图片并显示
首先将图片放到代码的同级目录下，通过imread载入，然后将其显示并打印出来
'''
import matplotlib.pyplot as plt # plt 用于显示图片
import matplotlib.image as mpimg # mpimg 用于读取图片
import numpy as np
import tensorflow as tf  
myimg = mpimg.imread('2.jpg') # 读取和代码处于同一目录下的图片
#myimg = mpimg.imread('img.jpg') # 读取和代码处于同一目录下的图片
plt.imshow(myimg) # 显示图片
plt.axis('off') # 不显示坐标轴
plt.show()
print(myimg.shape)
'''
上面这段代码输出(960, 720, 3)，可以看到，载入图片的维度是960*720大小，3个通道
'''
'''
这里需要手动将sobel算子填入卷积核里。使用tf.constant函数可以将常量直接初始化到Variable中，因为是3个通道，所以sobel卷积核的每个元素都扩成了3个。
注意：sobel算子处理过的图片并不保证每个像素都在0~25