图像中物体边缘检测

import matplotlib.pyplot as plt 
from PIL import Image
import numpy as np 
import paddle
import paddle.fluid as fluid
from paddle.fluid.dygraph.nn import Conv2D
from paddle.fluid.initializer import NumpyArrayInitializer


img=Image.open('./work/images/section1/000000098520.jpg')
with fluid.dygraph.guard():
  #设置卷积核参数
  w=np.array([[-1,-1,-1],[-1,8,-1],[-1,-1,-1]],dtype='float32')/8
  w=w.reshape([1,1,3,3])
  #彩色图片是RGB三个通道，将卷积核的形状变一下
  w=np.repeat(w,3,axis=1)
  #num_filters, 输出通道数目，相当于上文中的CoutC_{out}
  conv=Conv2D('conv',num_filters=1,filter_size=[3,3],param_attr=fluid.ParamAttr(initializer=NumpyArrayInitializer(value=w)))

  x=np.array(img).astype('float32')
  #图片读入ndarry时，形状是[H,W,3],需要将通道数调整到最前面
  x=np.transpose(x,(2,0,1))

  #数据调整为[N,C,H,W]
  x=x.reshape(1,3,img.height,img.width)
  x=fluid.dygraph.to_variable(x)
  y=conv(x)
  out=y.numpy()

plt.figure(figsize=(20,10))
f=plt.subplot(121)
f.set_title('input image',fontsize=15)
plt.imshow(img)
f=plt.subplot(122)
f.set_title('output featuremap',fontsize=15)
plt.imshow(out.squeeze(),cmap='gray')
plt.show()

原文链接：https://aistudio.baidu.com/bdvgpu/user/173710/240161/notebooks/240161.ipynb?redirects=1