-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfilter_gui.py
206 lines (181 loc) · 9.81 KB
/
filter_gui.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
"""
Shoreline Filters for CoastSeg (image suitability, segmentation filter, spatial KDE)
Mark Lundine, USGS
Requires tensorflow, pandas, numpy, pyqt, matplotlib, spatial-kde, rasterio, gdal, python=3.10
"""
#basic imports
import os
import glob
import sys
import shutil
#pyqt
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *
#filters
import image_segmentation_filter
import image_filter
import shoreline_change_envelope
class Window(QMainWindow):
def __init__(self):
super(Window, self).__init__()
sizeObject = QDesktopWidget().screenGeometry(-1)
global screenWidth
screenWidth = sizeObject.width()
global screenHeight
screenHeight = sizeObject.height()
global bw1
bw1 = int(screenWidth/15)
global bw2
bw2 = int(screenWidth/50)
global bh1
bh1 = int(screenHeight/15)
global bh2
bh2 = int(screenHeight/20)
self.setWindowTitle("CoastSeg Filters")
self.home()
def run_image_suitability_filter(self, threshold):
options = QFileDialog.Options()
options |= QFileDialog.DontUseNativeDialog
home = str(QFileDialog.getExistingDirectory(self, "Select RGB Image Folder"))
if home:
image_filter.run_inference_rgb(os.path.join(os.getcwd(), 'models', 'image_rgb', 'best.h5'),
home,
home,
os.path.join(home, 'good_bad.csv'),
threshold
)
def run_multi_image_suitability_filter(self, threshold):
options = QFileDialog.Options()
options |= QFileDialog.DontUseNativeDialog
home = str(QFileDialog.getExistingDirectory(self, "Select Sessions Folder"))
if home:
image_filter.inference_multiple_sessions(home, threshold, model='rgb')
def run_image_segmentation_filter(self, threshold):
options = QFileDialog.Options()
options |= QFileDialog.DontUseNativeDialog
home = str(QFileDialog.getExistingDirectory(self, "Select RGB Segmentation Image Folder"))
if home:
image_segmentation_filter.run_inference_rgb(os.path.join(os.getcwd(), 'models', 'segmentation_rgb', 'best_seg.h5'),
os.path.join(home),
os.path.join(home),
os.path.join(home, 'good_bad_seg.csv'),
threshold
)
def run_multi_image_segmentation_filter(self, threshold):
options = QFileDialog.Options()
options |= QFileDialog.DontUseNativeDialog
home = str(QFileDialog.getExistingDirectory(self, "Select Sessions Folder"))
if home:
image_segmentation_filter.inference_multiple_sessions(home, threshold)
def run_spatial_kde_filter(self, radius, cell_size, buffer):
options = QFileDialog.Options()
shorelines, _ = QFileDialog.getOpenFileName(self,"Select Extracted Shorelines Points GeoJSON", "","GeoJSON (*.geojson)", options=options)
if shorelines:
extracted_shorelines_points = shorelines
site = os.path.dirname(shorelines)
point_density_kde_path = os.path.join(site, 'spatial_kde.tif')
otsu_path = os.path.join(site, 'spatial_kde_otsu.tif')
shoreline_change_envelope_path = os.path.join(site, 'shoreline_change_envelope.geojson')
shoreline_change_envelope_buffer_path = os.path.join(site, 'shoreline_change_envelope_buffer.geojson')
shoreline_change_envelope.get_point_density_kde(extracted_shorelines_points,
point_density_kde_path,
otsu_path,
shoreline_change_envelope_path,
shoreline_change_envelope_buffer_path,
kde_radius=buffer,
cell_size=cell_size
)
def run_multi_spatial_kde_filter(self, radius, cell_size, buffer):
options = QFileDialog.Options()
options |= QFileDialog.DontUseNativeDialog
home = str(QFileDialog.getExistingDirectory(self, "Select Sessions Folder"))
if home:
get_point_density_kde_multiple_sessions(home,
kde_radius=80,
cell_size=15,
buffer=50)
def home(self):
self.scroll = QScrollArea() # Scroll Area which contains the widgets, set as the centralWidget
self.widget = QWidget() # Widget that contains the collection of Vertical Box
self.vbox = QGridLayout() # The Vertical Box that contains the Horizontal Boxes of labels and buttons
self.widget.setLayout(self.vbox)
#image suitability
image_suitability_filter = QPushButton('Image Suitability Filter Single Session')
self.vbox.addWidget(image_suitability_filter , 0, 0)
#image suitability
image_suitability_filter_multi = QPushButton('Image Suitability Filter Multiple Sessions')
self.vbox.addWidget(image_suitability_filter_multi , 1, 0)
#image suitability threshold
image_suitability_threshold_label = QLabel('Threshold')
image_suitability_threshold = QDoubleSpinBox()
image_suitability_threshold.setMinimum(0.01)
image_suitability_threshold.setMaximum(0.99)
image_suitability_threshold.setValue(0.437)
self.vbox.addWidget(image_suitability_threshold_label, 2, 0)
self.vbox.addWidget(image_suitability_threshold, 3, 0)
#segmentation filter button
segmentation_filter = QPushButton('Segmentation Filter Single Session')
self.vbox.addWidget(segmentation_filter, 0, 1)
#segmentation filter multiple button
segmentation_filter_multi = QPushButton('Segmentation Filter Multiple Sessions')
self.vbox.addWidget(segmentation_filter_multi, 1, 1)
#segmentation filter threshold
segmentation_filter_threshold_label = QLabel('Threshold')
segmentation_filter_threshold = QDoubleSpinBox()
segmentation_filter_threshold.setMinimum(0.01)
segmentation_filter_threshold.setMaximum(0.99)
segmentation_filter_threshold.setValue(0.482)
self.vbox.addWidget(segmentation_filter_threshold_label, 2, 1)
self.vbox.addWidget(segmentation_filter_threshold, 3, 1)
#spatial kde button
spatial_kde_filter = QPushButton('Spatial KDE Filter Single Session')
self.vbox.addWidget(spatial_kde_filter, 0, 2)
#spatial kde button
spatial_kde_filter_multi = QPushButton('Spatial KDE Filter Multiple Sessions')
self.vbox.addWidget(spatial_kde_filter_multi, 1, 2)
#spatial kde radiuds
radius_label = QLabel('Radius (m)')
radius_slider = QSpinBox()
radius_slider.setMinimum(1)
radius_slider.setMaximum(100)
radius_slider.setValue(80)
self.vbox.addWidget(radius_label, 2, 2)
self.vbox.addWidget(radius_slider, 3, 2)
#cell size
cell_size_label = QLabel('Cell Size (m)')
cell_size_slider = QSpinBox()
cell_size_slider.setMinimum(1)
cell_size_slider.setMaximum(50)
cell_size_slider.setValue(15)
self.vbox.addWidget(cell_size_label, 4, 2)
self.vbox.addWidget(cell_size_slider, 5, 2)
#buffer
buffer_label = QLabel('Buffer (m)')
buffer_slider = QSpinBox()
buffer_slider.setMinimum(0)
buffer_slider.setMaximum(50)
buffer_slider.setValue(15)
self.vbox.addWidget(buffer_label, 6, 2)
self.vbox.addWidget(buffer_slider, 7, 2)
#Actions
image_suitability_filter.clicked.connect(lambda: self.run_image_suitability_filter(image_suitability_threshold.value()))
image_suitability_filter_multi.clicked.connect(lambda: self.run_multi_image_suitability_filter(image_suitability_threshold.value()))
segmentation_filter.clicked.connect(lambda: self.run_image_segmentation_filter(segmentation_filter_threshold.value()))
segmentation_filter_multi.clicked.connect(lambda: self.run_multi_image_segmentation_filter(segmentation_filter_threshold.value()))
spatial_kde_filter.clicked.connect(lambda: self.run_spatial_kde_filter(radius_slider.value(), cell_size_slider.value(), buffer_slider.value()))
spatial_kde_filter_multi.clicked.connect(lambda: self.run_multi_spatial_kde_filter(radius_slider.value(), cell_size_slider.value(), buffer_slider.value()))
#Scroll policies
self.scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self.scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self.scroll.setWidgetResizable(True)
self.scroll.setWidget(self.widget)
self.setCentralWidget(self.scroll)
## Function outside of the class to run the app
def run():
app = QApplication(sys.argv)
GUI = Window()
GUI.show()
sys.exit(app.exec_())
## Calling run to run the app
run()