python - 从打印的心电图中提取图形

Question

我在堆栈上发现了一个这样的问题，但没有得到回答，而且我的图像不一样。

我有您在下面看到的这张图片。

我想提取图像上的图形。我已经尝试了很多方法来提取它。阈值处理似乎不起作用，因为 rgb 值非常接近，尤其是在图表的峰值处。我曾尝试使用渐变来检测图像上的流动，但也失败了。我尝试组合不同的颜色空间(HLS、YUV、LAB)它有一些输出，但它正在删除图形的某些部分。我还尝试创建 8X8 或更小的掩码矩阵，以获得平均像素值并删除低于该值的所有内容。还创建中值过滤。最后两个与前面提到的一些技术相结合有所帮助，但它们正在消除峰值，因为差异最小。你知道有什么更好的方法可以帮助我吗？

注:我用的是Python

Answer 1

你可以尝试对图像进行阈值处理 (HSV)

import cv2
import numpy as np
from PIL import Image
import PIL.ImageOps   

img=cv2.imread("ekg.jpg")
BLACK_MIN = np.array([0, 0, 0],np.uint8)
BLACK_MAX = np.array([255, 255, 80],np.uint8)
hsv_img = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
frame_threshed = cv2.inRange(hsv_img, BLACK_MIN, BLACK_MAX)
cv2.imwrite('threshed.jpg', frame_threshed)
image = Image.open('threshed.jpg')
inverted_image = PIL.ImageOps.invert(image)
inverted_image.save('invert.jpg')
img2=cv2.imread("invert.jpg")
cv2.imshow('image1', img2)