2023-09-14 14:24:12 +02:00
|
|
|
#!/usr/bin/python3
|
|
|
|
|
import pathlib
|
|
|
|
|
import pygubu
|
|
|
|
|
from tkinter import *
|
|
|
|
|
from PIL import ImageTk, Image
|
|
|
|
|
import numpy as np
|
|
|
|
|
import cv2
|
|
|
|
|
|
|
|
|
|
PROJECT_PATH = pathlib.Path(__file__).parent
|
|
|
|
|
PROJECT_UI = "./src/gui/main.ui"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MainApp:
|
|
|
|
|
def __init__(self, master=None):
|
|
|
|
|
self.builder = builder = pygubu.Builder()
|
|
|
|
|
builder.add_resource_path(PROJECT_PATH)
|
|
|
|
|
builder.add_from_file(PROJECT_UI)
|
|
|
|
|
# Main widget
|
|
|
|
|
self.mainwindow = builder.get_object("main", master)
|
|
|
|
|
|
|
|
|
|
self.canvas = builder.get_object("output_canvas")
|
|
|
|
|
self.tk_imgs = []
|
|
|
|
|
|
|
|
|
|
self.canny_thr1 = None
|
|
|
|
|
self.canny_thr2 = None
|
|
|
|
|
self.img_path = None
|
|
|
|
|
self.blur_rate = None
|
|
|
|
|
self.img_size = None
|
|
|
|
|
self.sobel_select = None
|
|
|
|
|
builder.import_variables(self,
|
|
|
|
|
['canny_thr1',
|
|
|
|
|
'canny_thr2',
|
|
|
|
|
'img_path',
|
|
|
|
|
'blur_rate',
|
|
|
|
|
'img_size',
|
|
|
|
|
'sobel_select'])
|
|
|
|
|
|
|
|
|
|
builder.connect_callbacks(self)
|
|
|
|
|
|
|
|
|
|
def run(self):
|
|
|
|
|
self.mainwindow.mainloop()
|
|
|
|
|
|
|
|
|
|
def update(self, event=None):
|
|
|
|
|
path = self.img_path.get()
|
|
|
|
|
|
|
|
|
|
if path != None and path != "":
|
|
|
|
|
# Get all user vars
|
|
|
|
|
ct1 = self.canny_thr1.get()
|
|
|
|
|
ct2 = self.canny_thr2.get()
|
|
|
|
|
br = self.blur_rate.get()
|
|
|
|
|
sxy = self.sobel_select.get()
|
|
|
|
|
size = self.img_size.get()
|
|
|
|
|
|
|
|
|
|
# Keep array of cv2 images to output
|
|
|
|
|
output = []
|
|
|
|
|
|
|
|
|
|
# Clear previously printed images
|
|
|
|
|
self.tk_imgs = []
|
|
|
|
|
|
|
|
|
|
# Import and resize image
|
|
|
|
|
img = cv2.imread(path)
|
|
|
|
|
img = cv2.resize(img, (size, size), interpolation = cv2.INTER_AREA)
|
|
|
|
|
output.append(img) # First output image is original
|
|
|
|
|
|
|
|
|
|
# Set grayscale
|
|
|
|
|
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
|
|
|
|
|
output.append(img_gray) # Second output is grayscale
|
|
|
|
|
|
|
|
|
|
# Blurred edition
|
|
|
|
|
if (br == 0):
|
|
|
|
|
# Disable
|
|
|
|
|
img_blur = img_gray
|
|
|
|
|
elif (br % 2 == 1):
|
|
|
|
|
img_blur = cv2.GaussianBlur(img_gray, (br, br), 0)
|
|
|
|
|
else:
|
|
|
|
|
print(f"Blur rate changed to {br - 1}")
|
|
|
|
|
img_blur = cv2.GaussianBlur(img_gray, (br-1, br-1), 0)
|
|
|
|
|
output.append(img_blur)
|
|
|
|
|
|
|
|
|
|
# Sobel edge
|
|
|
|
|
if sxy in ['x', 'y', 'both']:
|
|
|
|
|
if sxy == 'x':
|
|
|
|
|
dx = 1
|
|
|
|
|
dy = 0
|
|
|
|
|
elif sxy == 'y':
|
|
|
|
|
dx = 0
|
|
|
|
|
dy = 1
|
|
|
|
|
elif sxy == 'both':
|
|
|
|
|
dx = 1
|
|
|
|
|
dy = 1
|
|
|
|
|
|
|
|
|
|
img_sobel = cv2.Sobel(src=img_blur, ddepth=cv2.CV_8U, dx=dx, dy=dy, ksize=5)
|
|
|
|
|
else:
|
|
|
|
|
img_sobel = img_gray
|
|
|
|
|
output.append(img_sobel)
|
|
|
|
|
|
|
|
|
|
# Canny edge
|
|
|
|
|
img_canny = cv2.Canny(image=img_blur,threshold1=ct1,threshold2=ct2)
|
|
|
|
|
output.append(img_canny)
|
|
|
|
|
|
|
|
|
|
# Draw all output images
|
|
|
|
|
for idx, data in enumerate(output):
|
|
|
|
|
# Create ui image
|
|
|
|
|
tk_img = cv2.cvtColor(data, cv2.COLOR_BGR2RGB)
|
|
|
|
|
tk_img = ImageTk.PhotoImage(image=Image.fromarray(tk_img))
|
|
|
|
|
self.tk_imgs.append(tk_img)
|
|
|
|
|
|
|
|
|
|
self.canvas.create_image((size * idx),0,anchor=NW,image=self.tk_imgs[idx],tags="og")
|
|
|
|
|
|
|
|
|
|
self.canvas.draw()
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
app = MainApp()
|
|
|
|
|
app.run()
|
