卷积运算--Sobel算子边缘提取代码复现---python（tensorflow）

概述

卷积

从公式可以看出来对原函数平移并与核相乘，就是信号与系统中学的卷积运算。

在图像卷积中，即每个核的中心点对应的位置进行卷积，滑窗方法就是上面的卷积公式。

sobel算子是一种经典的边缘提取算子，以sobel为例进行卷积运算。

代码

# -*- coding: utf-8 -*-
"""
Created on Fri Apr 26 03:01:26 2019
sobel
@author: ybx
"""

import tensorflow as tf
import numpy as np
import cv2

image =cv2.imread('flower.jpg')#图像读取


gray = tf.cast(cv2.cvtColor(image, cv2.COLOR_BGR2GRAY),dtype = "float32")#转灰度图并转换为tensor型

image_x =tf.reshape(gray,[1,335,500,1])#变换维度（batch，height，weight，channel）