python对图像提取边缘，采用canny算子

版权声明：我是南七小僧，微信: to_my_love ，寻找人工智能相关工作，欢迎交流思想碰撞。 https://blog.csdn.net/qq_25439417/article/details/79729370

import tensorflow as tf
import numpy as np
import skimage as si
import tensorflow.examples.tutorials.mnist.input_data as input_data
import matplotlib.pyplot as plt
from skimage import data,filters,feature
import os


os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'


mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
batch=mnist.train.next_batch(1)
x=batch[0]
print(batch[0])
x_img=np.reshape(x,(28,28))
plt.subplot(121)
plt.imshow(x_img)
x_img=feature.canny(x_img,sigma=0.5)
plt.subplot(122)
plt.imshow(x_img)