1. 使用OpenCV库进行边缘检测和背景去除。 示例代码：

import cv2

读取图像

img = cv2.imread('image.jpg')

转为灰度图像

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

Canny边缘检测

edges = cv2.Canny(gray, 100, 200)

去除背景

fgmask = cv2.createBackgroundSubtractorMOG2().apply(img) edges[fgmask == 0] = 0

显示结果

cv2.imshow('Edges', edges) cv2.waitKey(0)

2. 使用深度学习模型进行边缘检测和背景去除。 示例代码：

import cv2 import numpy as np import tensorflow as tf

读取图像

img = cv2.imread('image.jpg')

加载深度学习模型

model = tf.keras.models.load_model('model.h5')

预处理图像

img = cv2.resize(img, (256, 256)) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img = np.expand_dims(img, 0) img = img / 255.0

获取模型输出

edges, fgmask = model.predict(img)

去除背景

edges = np.argmax(edges, axis=-1)[0] edges[fgmask == 0] = 0

显示结果

cv2.imshow('Edges', edges) cv2.waitKey(0)