写一段去除复杂背景的python代码
时间: 2023-04-08 22:04:50 浏览: 140
使用 OpenCV 和深度学习 去除图像的背景_python_代码_下载
5星 · 资源好评率100%
以下是一段去除复杂背景的Python代码:
```python
import cv2
# 读取图片
img = cv2.imread('image.jpg')
# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 进行高斯模糊
blur = cv2.GaussianBlur(gray, (5, 5), 0)
# 进行二值化处理
ret, thresh = cv2.threshold(blur, 0, 255, cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
# 进行形态学操作
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5, 5))
closing = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel)
# 查找轮廓
contours, hierarchy = cv2.findContours(closing, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# 找到最大的轮廓
max_contour = max(contours, key=cv2.contourArea)
# 创建掩模
mask = np.zeros(img.shape[:2], np.uint8)
cv2.drawContours(mask, [max_contour], 0, 255, -1)
# 应用掩模
res = cv2.bitwise_and(img, img, mask=mask)
# 显示结果
cv2.imshow('Result', res)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
这段代码可以去除图片中的复杂背景,只保留主体部分。
阅读全文