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

最新推荐文章于 2024-06-11 16:41:22 发布

hfut--罗小黑

最新推荐文章于 2024-06-11 16:41:22 发布

阅读量1.3k

点赞数 1

本文链接：https://blog.csdn.net/qq_37588821/article/details/89609081

版权

本文介绍了使用Sobel算子进行边缘提取的原理，并通过Python和Tensorflow展示了具体代码实现。卷积过程通过滑窗方法完成，结果显示花的边缘明显。代码具有一定的通用性，可以替换核来应用其他边缘提取算子。

摘要由CSDN通过智能技术生成

概述

卷积

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

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

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）

kernal = tf.constant([[-1,0,1],
                      [-2,0,2],
                      [-1,0,1]],dtype="f