Initial commit
This commit is contained in:
226
app/app.py
Executable file
226
app/app.py
Executable file
@ -0,0 +1,226 @@
|
||||
import tkinter as tk
|
||||
from pathlib import Path
|
||||
from tkinter import filedialog, messagebox
|
||||
|
||||
import cv2
|
||||
import ttkbootstrap as ttk
|
||||
from convert import differences, load_frames, select_frames
|
||||
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk
|
||||
from matplotlib.figure import Figure
|
||||
from params import ParameterTuner
|
||||
from PIL import Image, ImageTk
|
||||
from preview import SlidePreview
|
||||
|
||||
|
||||
class SlideConversionApp:
|
||||
def __init__(self):
|
||||
self._root = tk.Tk()
|
||||
self._root.title("PDF Extraktion")
|
||||
self._root.geometry("1400x1000")
|
||||
|
||||
self._frames = None
|
||||
self._diffs = None
|
||||
|
||||
# file selection
|
||||
self._file_selection = tk.Frame(self._root)
|
||||
self._file_selection_label = tk.Label(self._file_selection, text="Videodatei:")
|
||||
self._file_selection_label.pack()
|
||||
self._file_selection_button = ttk.Button(
|
||||
self._file_selection, text="Datei auswählen", command=self._select_file
|
||||
)
|
||||
self._file_selection_button.pack()
|
||||
self._file_selection.pack()
|
||||
self._file = None
|
||||
ttk.Separator(self._root, orient=tk.HORIZONTAL).pack(fill=tk.X, pady=10, padx=10)
|
||||
|
||||
# slide selection
|
||||
self._selection = tk.Frame(self._root)
|
||||
self._params = ParameterTuner(self._selection)
|
||||
self._param_threshold = self._params.add("Schwellenwert", 10, 1, 20)
|
||||
self._params.pack(side=tk.LEFT, padx=10)
|
||||
self._show_diffs_btn = ttk.Button(self._selection, text="Unterschiede anzeigen", command=self._show_diffs)
|
||||
self._show_diffs_btn.pack(side=tk.LEFT, padx=10)
|
||||
self._add_selected_btn = ttk.Button(self._selection, text="Folien hinzufügen", command=self._add_selection)
|
||||
self._add_selected_btn.pack(side=tk.LEFT, padx=10)
|
||||
self._selection.pack()
|
||||
ttk.Separator(self._root, orient=tk.HORIZONTAL).pack(fill=tk.X, pady=10, padx=10)
|
||||
|
||||
# generated images
|
||||
self._images = SlidePreview(self._root, columns=6)
|
||||
self._images.pack(fill=tk.BOTH, padx=10, expand=True)
|
||||
ttk.Separator(self._root, orient=tk.HORIZONTAL).pack(fill=tk.X, pady=10, padx=10)
|
||||
|
||||
# buttons
|
||||
self._buttons = tk.Frame(self._root)
|
||||
self._buttons.pack(side=tk.RIGHT, pady=(0, 5))
|
||||
self._button_create_pdf = ttk.Button(self._buttons, text="PDF erstellen", command=self._create_pdf)
|
||||
self._button_create_pdf.pack(side=tk.RIGHT, padx=10)
|
||||
self._button_add_frame = ttk.Button(self._buttons, text="Manuell hinzufügen", command=self._add_manually)
|
||||
self._button_add_frame.pack(side=tk.RIGHT, padx=10)
|
||||
|
||||
self._set_interaction_state(tk.DISABLED)
|
||||
|
||||
self._root.mainloop()
|
||||
|
||||
def _select_file(self):
|
||||
file = filedialog.askopenfilename(
|
||||
title="Videodatei auswählen",
|
||||
filetypes=[
|
||||
("Videodateien", "*.mp4"),
|
||||
("Alle Dateien", "*.*"),
|
||||
],
|
||||
)
|
||||
if file is not None and file != "":
|
||||
file = Path(file)
|
||||
self._file_selected(file)
|
||||
|
||||
def _file_selected(self, file: Path):
|
||||
# set variables
|
||||
cap = cv2.VideoCapture(file.as_posix())
|
||||
self._frames = None
|
||||
|
||||
# analyze file
|
||||
fps = int(cap.get(cv2.CAP_PROP_FPS))
|
||||
num_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
||||
|
||||
# load frames
|
||||
status = tk.Toplevel(self._root)
|
||||
status.title("Lade Videoframes ..")
|
||||
status.grab_set()
|
||||
status_label = tk.Label(status, text=f"Lade Videoframes (0/{num_frames})")
|
||||
status_label.pack(padx=10, pady=10)
|
||||
status_pb = ttk.Progressbar(status, bootstyle="progress", orient=tk.HORIZONTAL, length=200, mode="determinate")
|
||||
status_pb.pack(padx=10, pady=10)
|
||||
|
||||
def callback(text: str, frame_no: int, max_no: int):
|
||||
status_label.configure(text=f"{text} ({frame_no}/{max_no})")
|
||||
status_pb.configure(value=int((frame_no / max_no) * 100))
|
||||
status_label.update()
|
||||
status_pb.update()
|
||||
status_pb.update_idletasks()
|
||||
|
||||
frames = load_frames(cap, fps, lambda f: callback("Lade Videoframes", f, num_frames))
|
||||
|
||||
diffs = differences(frames, lambda f: callback("Berechne Unterschiede", f, len(frames)))
|
||||
|
||||
status.grab_release()
|
||||
status.destroy()
|
||||
|
||||
# update values & parameters
|
||||
self._frames = frames
|
||||
self._diffs = diffs
|
||||
self._params.get_slider(self._param_threshold).configure(to=int(max(diffs)))
|
||||
self._file_selection_button.configure(text=file.name)
|
||||
self._file = file
|
||||
self._images.clear()
|
||||
self._set_interaction_state(tk.NORMAL)
|
||||
|
||||
def _set_interaction_state(self, state: str):
|
||||
elements = [self._button_create_pdf, self._show_diffs_btn, self._add_selected_btn, self._button_add_frame]
|
||||
|
||||
self._params.set_state(state)
|
||||
for element in elements:
|
||||
element.configure(state=state)
|
||||
|
||||
def _show_diffs(self):
|
||||
if self._diffs is None:
|
||||
return
|
||||
|
||||
diffs = tk.Toplevel(self._root)
|
||||
diffs.title("Unterschiede zwischen Frames")
|
||||
diffs.grab_set()
|
||||
fig = Figure(figsize=(10, 5), dpi=100)
|
||||
plot = fig.add_subplot(111)
|
||||
plot.plot(self._diffs)
|
||||
canvas = FigureCanvasTkAgg(fig, master=diffs)
|
||||
canvas.draw()
|
||||
canvas.get_tk_widget().pack()
|
||||
toolbar = NavigationToolbar2Tk(canvas, diffs)
|
||||
toolbar.update()
|
||||
canvas.get_tk_widget().pack()
|
||||
|
||||
def _add_selection(self):
|
||||
self._images.clear()
|
||||
images = select_frames(self._frames, self._diffs, threshold=self._params.get(self._param_threshold))
|
||||
for image in images:
|
||||
self._images.add(image)
|
||||
|
||||
def _create_pdf(self):
|
||||
if len(self._images) == 0:
|
||||
messagebox.showerror(title="PDF erstellen", message="Keine Folien vorhanden.")
|
||||
return
|
||||
|
||||
file = filedialog.asksaveasfilename(
|
||||
title="Speichern Unter",
|
||||
filetypes=[
|
||||
("PDF", "*.pdf"),
|
||||
("Alle Dateien", "*.*"),
|
||||
],
|
||||
)
|
||||
|
||||
if file is not None and file != "":
|
||||
file = Path(file).with_suffix(".pdf")
|
||||
else:
|
||||
return
|
||||
|
||||
images = self._images.get_images()
|
||||
images[0].save(file, "PDF", resolution=100.0, save_all=True, append_images=images[1:])
|
||||
messagebox.showinfo(title="PDF erstellen", message="Erfolgreich!")
|
||||
|
||||
def _add_manually(self):
|
||||
if self._images._selected_index == -1:
|
||||
messagebox.showerror(
|
||||
title="Manuelles Hinzufügen",
|
||||
message="Wähle zunächst eine Folie zum dahinter einfügen aus",
|
||||
)
|
||||
return
|
||||
|
||||
frameview = tk.Toplevel(self._root)
|
||||
frameview.title("Manuell Hinzufügen")
|
||||
frameview.grab_set()
|
||||
|
||||
thumbnail = tk.Label(frameview)
|
||||
thumbnail.pack(padx=10, pady=5)
|
||||
|
||||
def show(frame_no: int):
|
||||
img = Image.fromarray(self._frames[frame_no][:, :, ::-1])
|
||||
img.thumbnail(size=(512, 512))
|
||||
thumbnail.current = ImageTk.PhotoImage(img)
|
||||
thumbnail.configure(image=thumbnail.current)
|
||||
|
||||
show(0)
|
||||
|
||||
selected = tk.IntVar(value=0)
|
||||
selected.trace_add("write", lambda v, i, m: show(selected.get()))
|
||||
|
||||
framesel = tk.Frame(frameview)
|
||||
framesel.pack(padx=10, pady=5)
|
||||
back_btn = ttk.Button(
|
||||
framesel,
|
||||
text="<",
|
||||
command=lambda: selected.set(0 if selected.get() - 1 < 0 else selected.get() - 1),
|
||||
)
|
||||
next_btn = ttk.Button(
|
||||
framesel,
|
||||
text=">",
|
||||
command=lambda: selected.set(
|
||||
len(self._frames) - 1 if selected.get() + 1 >= len(self._frames) else selected.get() + 1
|
||||
),
|
||||
)
|
||||
slider = ttk.Scale(framesel, from_=0, to=len(self._frames) - 1, variable=selected, length=300)
|
||||
back_btn.pack(side=tk.LEFT)
|
||||
slider.pack(side=tk.LEFT, padx=5)
|
||||
next_btn.pack(side=tk.LEFT)
|
||||
|
||||
def add_image():
|
||||
self._images.insert(
|
||||
Image.fromarray(self._frames[selected.get()][:, :, ::-1]),
|
||||
self._images._selected_index + 1,
|
||||
)
|
||||
|
||||
add_btn = ttk.Button(frameview, text="Hinzufügen", command=add_image)
|
||||
add_btn.pack(pady=5)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
app = SlideConversionApp()
|
||||
50
app/convert.py
Executable file
50
app/convert.py
Executable file
@ -0,0 +1,50 @@
|
||||
from typing import Callable
|
||||
|
||||
import cv2
|
||||
import numpy as np
|
||||
from PIL import Image
|
||||
from tqdm import tqdm, trange
|
||||
|
||||
|
||||
def compare(frame1, frame2):
|
||||
absdiff = cv2.absdiff(frame1, frame2)
|
||||
return np.mean(absdiff)
|
||||
|
||||
|
||||
def load_frames(cap: cv2.VideoCapture, frame_interval: int, progress: Callable) -> list:
|
||||
frames = []
|
||||
num_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
|
||||
|
||||
i = 0
|
||||
res = cap.grab()
|
||||
while i < num_frames:
|
||||
if res:
|
||||
frames.append(cap.retrieve()[1])
|
||||
for _ in range(frame_interval):
|
||||
res = cap.grab()
|
||||
i += frame_interval
|
||||
progress(i)
|
||||
|
||||
return frames
|
||||
|
||||
|
||||
def differences(frames: list, progress: Callable) -> list[float]:
|
||||
diffs = []
|
||||
|
||||
for index, (frame1, frame2) in enumerate(zip(frames[:-1], frames[1:]), start=1):
|
||||
diff = compare(frame1, frame2)
|
||||
diffs.append(diff)
|
||||
progress(index)
|
||||
|
||||
return diffs
|
||||
|
||||
|
||||
def select_frames(frames: list, diffs: list, threshold: float) -> list[Image.Image]:
|
||||
selected_frames = [frames[0]]
|
||||
|
||||
for frame, diff in zip(frames[1:], diffs, strict=True):
|
||||
if diff > threshold:
|
||||
selected_frames.append(frame)
|
||||
|
||||
# Convert to Images
|
||||
return [Image.fromarray(frame[:, :, ::-1]) for frame in selected_frames]
|
||||
77
app/params.py
Executable file
77
app/params.py
Executable file
@ -0,0 +1,77 @@
|
||||
import tkinter as tk
|
||||
from dataclasses import dataclass
|
||||
|
||||
import ttkbootstrap as ttk
|
||||
|
||||
|
||||
@dataclass
|
||||
class Parameter:
|
||||
var: tk.IntVar | tk.DoubleVar
|
||||
slider: ttk.Scale
|
||||
|
||||
|
||||
class ParameterTuner(tk.Frame):
|
||||
def __init__(self, parent, *args, **kwargs):
|
||||
super().__init__(parent, *args, *kwargs)
|
||||
|
||||
self._params: dict[int, Parameter] = {}
|
||||
|
||||
def add(self, name: str, value: int | float, min_value: int | float, max_value: int | float) -> int:
|
||||
frame = tk.Frame(self, width=200)
|
||||
|
||||
if isinstance(value, int):
|
||||
var = tk.IntVar(frame, value=value)
|
||||
labelformat = "{name}: {value}"
|
||||
else:
|
||||
var = tk.DoubleVar(frame, value=value)
|
||||
labelformat = "{name}: {value:.2f}"
|
||||
|
||||
label = tk.Label(frame, text=f"{name}: {var.get()}")
|
||||
slider = ttk.Scale(
|
||||
frame,
|
||||
variable=var,
|
||||
from_=min_value,
|
||||
to=max_value,
|
||||
length=200,
|
||||
orient=tk.HORIZONTAL,
|
||||
)
|
||||
|
||||
var.trace_add(
|
||||
"write",
|
||||
lambda v, index, mode: label.configure(text=labelformat.format(name=name, value=var.get())),
|
||||
)
|
||||
|
||||
label.pack()
|
||||
slider.pack()
|
||||
|
||||
frame.pack(side=tk.RIGHT, padx=10)
|
||||
|
||||
param_id = len(self._params)
|
||||
self._params[param_id] = Parameter(var, slider)
|
||||
return param_id
|
||||
|
||||
def get(self, param_id: int) -> int | float:
|
||||
return self._params[param_id].var.get()
|
||||
|
||||
def set(self, param_id: int, value: int | float):
|
||||
self._params[param_id].var.set(value)
|
||||
|
||||
def get_slider(self, param_id: int) -> ttk.Scale:
|
||||
return self._params[param_id].slider
|
||||
|
||||
def set_state(self, state: str, param_id: int = None):
|
||||
if param_id is not None:
|
||||
self._params[param_id].slider.configure(state=state)
|
||||
else:
|
||||
for param in self._params.values():
|
||||
param.slider.configure(state=state)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main = tk.Tk()
|
||||
pt = ParameterTuner(main)
|
||||
param1 = pt.add("Frameabstand", 30, 1, 60)
|
||||
param2 = pt.add("Schwellenwert", 0.2, 0, 5)
|
||||
pt.pack()
|
||||
|
||||
main.mainloop()
|
||||
151
app/preview.py
Executable file
151
app/preview.py
Executable file
@ -0,0 +1,151 @@
|
||||
import tkinter as tk
|
||||
from dataclasses import dataclass
|
||||
|
||||
import ttkbootstrap as ttk
|
||||
from PIL import Image, ImageTk
|
||||
from ttkbootstrap.scrolled import ScrolledFrame
|
||||
|
||||
|
||||
class Slide(tk.Frame):
|
||||
def __init__(self, preview, image: Image.Image):
|
||||
super().__init__(preview._frame)
|
||||
self.preview = preview
|
||||
|
||||
self.image = image
|
||||
self.imagetk = ImageTk.PhotoImage(image)
|
||||
|
||||
self.label = tk.Label(self, image=self.imagetk, borderwidth=1, relief=tk.SOLID)
|
||||
self.label.pack(fill=tk.X, pady=(0, 5))
|
||||
|
||||
self.delbutton = ttk.Button(self, text="Löschen", bootstyle="danger-outline", command=self._on_delete)
|
||||
self.delbutton.pack(fill=tk.X)
|
||||
|
||||
self._is_selected = False
|
||||
self.highlight = None
|
||||
|
||||
self._configure_id = None
|
||||
self.bind("<Configure>", self._on_resize)
|
||||
self.label.bind("<Button-1>", self._on_select)
|
||||
|
||||
def _on_resize(self, e: tk.Event):
|
||||
if self._configure_id != None:
|
||||
self.after_cancel(self._configure_id)
|
||||
self._configure_id = self.after(100, self._on_final_resize)
|
||||
|
||||
def _on_select(self, e: tk.Event):
|
||||
if self._is_selected:
|
||||
self._on_unselect()
|
||||
else:
|
||||
self._is_selected = True
|
||||
self.highlight = tk.Label(self, text="Ausgewählt", bg="green")
|
||||
self.highlight.place(x=0, y=0)
|
||||
self.preview.notify_selection(self)
|
||||
|
||||
def _on_unselect(self):
|
||||
self._is_selected = False
|
||||
if self.highlight is not None:
|
||||
self.highlight.destroy()
|
||||
|
||||
def _on_final_resize(self):
|
||||
thumbnail = self.image.copy()
|
||||
thumbnail.thumbnail(size=(self.winfo_width(), self.image.height), resample=Image.NEAREST)
|
||||
self.imagetk = ImageTk.PhotoImage(thumbnail)
|
||||
|
||||
self.label.configure(image=self.imagetk)
|
||||
self._configure_id = None
|
||||
|
||||
def _on_delete(self):
|
||||
self.preview.remove(self)
|
||||
|
||||
|
||||
class SlidePreview(ScrolledFrame):
|
||||
def __init__(self, parent, *args, columns: int = 4, **kwargs):
|
||||
super().__init__(parent, *args, **kwargs)
|
||||
|
||||
self._frame = tk.Frame(self)
|
||||
self._frame.pack(fill=tk.BOTH, expand=True)
|
||||
|
||||
self.columns = columns
|
||||
for column in range(columns):
|
||||
self._frame.grid_columnconfigure(column, weight=1)
|
||||
|
||||
self._selected_index = -1
|
||||
self._slides = []
|
||||
|
||||
def _on_resize(self, e: tk.Event):
|
||||
slide = self._slides[0]
|
||||
slide.grid_forget()
|
||||
slide.grid(row=0, column=0, sticky=tk.EW)
|
||||
|
||||
def notify_selection(self, selected: Slide):
|
||||
self._selected_index = self._slides.index(selected)
|
||||
for slide in self._slides:
|
||||
if slide is not selected:
|
||||
slide._on_unselect()
|
||||
|
||||
def notify_unselected(self, unselected: Slide):
|
||||
if self._slides.index(unselected) == self._selected_index:
|
||||
self._selected_index = -1
|
||||
|
||||
def add(self, slide: Image.Image):
|
||||
index = len(self._slides)
|
||||
|
||||
s = Slide(self, slide)
|
||||
s.grid(row=index // self.columns, column=index % self.columns, sticky=tk.NSEW, padx=10, pady=10)
|
||||
|
||||
self._slides.append(s)
|
||||
|
||||
def insert(self, slide: Image.Image, index: int):
|
||||
s = Slide(self, slide)
|
||||
self._slides.insert(index, s)
|
||||
|
||||
for index, slide in enumerate(self._slides):
|
||||
slide.grid_forget()
|
||||
slide.grid(row=index // self.columns, column=index % self.columns, sticky=tk.NSEW, padx=10, pady=10)
|
||||
slide._on_final_resize()
|
||||
|
||||
if index <= self._selected_index:
|
||||
self._selected_index += 1
|
||||
|
||||
def remove(self, slide: Slide):
|
||||
index = self._slides.index(slide)
|
||||
if index < 0:
|
||||
return
|
||||
|
||||
if index == self._selected_index:
|
||||
self._selected_index = -1
|
||||
elif index < self._selected_index:
|
||||
self._selected_index -= 1
|
||||
|
||||
slide.destroy()
|
||||
del self._slides[index]
|
||||
|
||||
for index, slide in enumerate(self._slides[index:], start=index):
|
||||
slide.grid_forget()
|
||||
slide.grid(row=index // self.columns, column=index % self.columns, sticky=tk.NSEW, padx=10, pady=10)
|
||||
|
||||
def clear(self):
|
||||
for slide in self._slides:
|
||||
slide.destroy()
|
||||
self._slides = []
|
||||
self._selected_index = -1
|
||||
|
||||
def __len__(self):
|
||||
return len(self._slides)
|
||||
|
||||
def get_images(self) -> list[Image.Image]:
|
||||
return [slide.image for slide in self._slides]
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main = tk.Tk()
|
||||
main.geometry("800x400")
|
||||
|
||||
sp = SlidePreview(main)
|
||||
sp.pack(fill=tk.BOTH, expand=True)
|
||||
|
||||
for filename in [f"{i:02}" for i in range(16)]:
|
||||
img = Image.open(f"machine learning in der radiologie_5ebbrbz/{filename}.png")
|
||||
sp.add(img)
|
||||
|
||||
main.mainloop()
|
||||
4
app/requirements.txt
Executable file
4
app/requirements.txt
Executable file
@ -0,0 +1,4 @@
|
||||
opencv-python
|
||||
ttkbootstrap
|
||||
tqdm
|
||||
matplotlib
|
||||
Reference in New Issue
Block a user