Compare commits
No commits in common. "27312ae6df03700cd52185d85db6899951fd8ef8" and "00e35ad78e52c8b9e4ee5ad7682125f3715149fa" have entirely different histories.
27312ae6df
...
00e35ad78e
34
src/suite.py
34
src/suite.py
@ -92,32 +92,7 @@ class MainApp:
|
|||||||
self.axs[column, row].plot(hist, label=lab)
|
self.axs[column, row].plot(hist, label=lab)
|
||||||
self.axs[column, row].grid()
|
self.axs[column, row].grid()
|
||||||
self.axs[column, row].legend()
|
self.axs[column, row].legend()
|
||||||
|
|
||||||
def drawCannyHM(self, img, column, row):
|
|
||||||
self.axs[column, row].clear()
|
|
||||||
canny_max = 500
|
|
||||||
canny_step = 20
|
|
||||||
|
|
||||||
results = [[] for x in range((int)(canny_max / canny_step))]
|
|
||||||
|
|
||||||
for th1 in range(0, canny_max, canny_step):
|
|
||||||
for th2 in range(0, canny_max, canny_step):
|
|
||||||
# Canny Edge Detection
|
|
||||||
edges = cv2.Canny(image=img, threshold1=th1, threshold2=th2)
|
|
||||||
|
|
||||||
w_res = cv2.countNonZero(edges)
|
|
||||||
y_ind = (int)(th1 / canny_step)
|
|
||||||
x_ind = (int)(th2 / canny_step)
|
|
||||||
|
|
||||||
results[y_ind].append(w_res)
|
|
||||||
|
|
||||||
# print(f"Result at thres {th1}, {th2}; \tIndex {y_ind}, {x_ind} \t= {w_res}")
|
|
||||||
# print(results[y_ind])
|
|
||||||
|
|
||||||
self.axs[column, row].imshow(results)
|
|
||||||
self.axs[column, row].xaxis.set_major_formatter(lambda x, pos: str(x*canny_step))
|
|
||||||
self.axs[column, row].yaxis.set_major_formatter(lambda x, pos: str(x*canny_step))
|
|
||||||
|
|
||||||
def update(self, event=None):
|
def update(self, event=None):
|
||||||
path = self.img_path.get()
|
path = self.img_path.get()
|
||||||
print(path)
|
print(path)
|
||||||
@ -136,7 +111,10 @@ class MainApp:
|
|||||||
|
|
||||||
# Get all user vars
|
# Get all user vars
|
||||||
ct1 = self.canny_thr1.get()
|
ct1 = self.canny_thr1.get()
|
||||||
ct2 = self.canny_thr2.get()
|
ct2 = self.canny_thr2.get()
|
||||||
|
|
||||||
|
print(f"canny 1: {ct1}, canny2: {ct2}")
|
||||||
|
|
||||||
br = self.blur_rate.get()
|
br = self.blur_rate.get()
|
||||||
sxy = self.sobel_select.get()
|
sxy = self.sobel_select.get()
|
||||||
size = self.img_size.get()
|
size = self.img_size.get()
|
||||||
@ -203,9 +181,7 @@ class MainApp:
|
|||||||
output.append(img_hsv[:, :, 2]) # V
|
output.append(img_hsv[:, :, 2]) # V
|
||||||
if img_hsv is not None:
|
if img_hsv is not None:
|
||||||
self.drawHist(img_hsv, ('H', 'S', 'V'), 0, 1)
|
self.drawHist(img_hsv, ('H', 'S', 'V'), 0, 1)
|
||||||
|
|
||||||
self.drawCannyHM(img, 1, 1)
|
|
||||||
|
|
||||||
plt.show(block=False)
|
plt.show(block=False)
|
||||||
|
|
||||||
# Draw all output images
|
# Draw all output images
|
||||||
|
Loading…
Reference in New Issue
Block a user