Python 如何在FFT幅度谱上绘制掩模线

嗨，我有一幅图像经过FFT运算后的幅度谱，现在我想画一个遮罩。然而，使用霍夫线等，我得到了太多的线和面具失败。下面是我需要创建的图像和遮罩的链接

提前谢谢

这是密码

import cv2
import numpy as np
from matplotlib import pyplot as plt
from scipy.misc import imshow, imsave, imread ,imsave
import mahotas


scaler = .5
img = imread("color2.jpeg")[:,:,:3]
mask = imread("image4.jpg")[:,:,:3]

img = cv2.resize(img, (0,0), fx=scaler, fy=scaler)
imggray = np.mean(img, -1)
imfft = np.fft.fft2(imggray)
mags = np.abs(np.fft.fftshift(imfft))
angles = np.angle(np.fft.fftshift(imfft))
visual = np.log(mags)
visual2 = (visual - visual.min()) / (visual.max() - visual.min())*255
cv2.imshow('Visual 2', visual2.astype(np.uint8))

imsave('fftimg4.jpg',visual2)

height,width,depth = img.shape
masking = np.zeros((height,width))

vis = visual2.astype(np.uint8)
edges = cv2.Canny(vis,50,180)
cv2.imshow('Gradient', edges)

lines = cv2.HoughLines(edges,1,np.pi/180,200,200,10)
for rho,theta in lines[0]:
   a = np.cos(theta)
   b = np.sin(theta)
   x0 = a*rho
   y0 = b*rho
   x1 = int(x0 + 1000*(-b))
   y1 = int(y0 + 1000*(a))
   x2 = int(x0 - 1000*(-b))
   y2 = int(y0 - 1000*(a))

   cv2.line(masking,(x1,y1),(x2,y2),(255,255,255),1)

   cv2.imshow('HoughLines', masking)

   mask = (np.mean(mask,-1) > 20)
   visual[mask] = np.mean(visual)
   newmagsshift = np.exp(visual)
   newffts = newmagsshift * np.exp(1j*angles)
   newfft = np.fft.ifftshift(newffts)
   imrev = np.fft.ifft2(newfft)
   newim2 = 255 - np.abs(imrev).astype(np.uint8)
   imsave("fftimg2.jpg",  newim2  )

   cv2.imshow('Image without Lines', newim2.astype(np.uint8))

cv2.waitKey(0)

---

首先，请澄清。在fft中，至少有三条“垂直”线和一堆“水平”线，对吗？另外，您是否真的需要自动执行此操作，或者此筛选操作可能需要手动完成？你的hough线算法会给你一系列的线方程吗

---

看起来你可以过滤掉那些穿过零的线，因为这就是你所期望的。穿过零度，也最接近预期角度，如示例中的-5度。

您好，欢迎来到Stack Overflow，如果您明确说明问题并提供源代码显示您尝试解决方案，您将获得更好的效果。也许可以尝试降低Hough空间的粒度，以便获得更少的行输出。因此，与其说是1度的粒度，不如说在Hough空间中有更少的对应于3度或5度的容器。注意@msanti这里是原始图像，更准确地说是fft2。我已经将代码添加到原始消息中，但仍然没有回答以下问题：1它应该是多么自动化，您能容忍用户从选项中选择正确的掩码吗？2如果不是，不同样本的2D FFT有多相似，您是否可以预先选择类型，例如，您真正想要的线条角度？2.1您为什么不从选项中选择最长的一行并使用它呢？它应该是完全自动化的，因为这会输入到另一个模块中。不，它不能是用户驱动器，因为我正在使用它来删除纸张上存在一些图像的规则线。扫描的每一页的遮罩都会不同，如果运行我的代码，您将得到多条线，这些线在不同角度上一样长