coralnet-toolbox 0.0.68__py2.py3-none-any.whl → 0.0.69__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,176 @@
1
+ import os
2
+ import glob
3
+ import sqlite3
4
+ import warnings
5
+
6
+ import faiss
7
+
8
+ warnings.filterwarnings("ignore", category=DeprecationWarning)
9
+
10
+
11
+ # ----------------------------------------------------------------------------------------------------------------------
12
+ # Viewers
13
+ # ----------------------------------------------------------------------------------------------------------------------
14
+
15
+
16
+ class FeatureStore:
17
+ """
18
+ Manages storing and retrieving annotation features for MULTIPLE models
19
+ using a single SQLite database and multiple, model-specific FAISS indexes.
20
+ """
21
+ def __init__(self, db_path='feature_store.db', index_path_base='features'):
22
+ self.db_path = db_path
23
+ self.index_path_base = index_path_base # Base name for index files, e.g., 'features'
24
+ self.conn = sqlite3.connect(self.db_path)
25
+ self.cursor = self.conn.cursor()
26
+ self._create_table()
27
+
28
+ # A dictionary to hold multiple FAISS indexes, keyed by model_key
29
+ self.faiss_indexes = {}
30
+
31
+ def _create_table(self):
32
+ """Create the metadata table if it doesn't exist."""
33
+ self.cursor.execute('''
34
+ CREATE TABLE IF NOT EXISTS features (
35
+ annotation_id TEXT NOT NULL,
36
+ model_key TEXT NOT NULL,
37
+ faiss_index INTEGER NOT NULL,
38
+ PRIMARY KEY (annotation_id, model_key)
39
+ )
40
+ ''')
41
+ self.conn.commit()
42
+
43
+ def _get_or_load_index(self, model_key):
44
+ """
45
+ Retrieves an index from memory or loads it from disk if it exists.
46
+ Returns the index object or None if not found in memory or on disk.
47
+ """
48
+ # 1. Check if the index is already loaded in memory
49
+ if model_key in self.faiss_indexes:
50
+ return self.faiss_indexes[model_key]
51
+
52
+ # 2. If not in memory, check for a corresponding file on disk
53
+ index_path = f"{self.index_path_base}_{model_key}.faiss"
54
+ if os.path.exists(index_path):
55
+ print(f"Loading existing FAISS index from {index_path}")
56
+ index = faiss.read_index(index_path)
57
+ self.faiss_indexes[model_key] = index # Cache it in memory
58
+ return index
59
+
60
+ # 3. If not in memory or on disk, return None
61
+ return None
62
+
63
+ def add_features(self, data_items, features, model_key):
64
+ """
65
+ Adds new features to the store for a specific model.
66
+ """
67
+ if not len(features):
68
+ return
69
+
70
+ # Get the specific index for this model, loading it if necessary
71
+ index = self._get_or_load_index(model_key)
72
+
73
+ # If no index exists yet, create one
74
+ if index is None:
75
+ feature_dim = features.shape[1]
76
+ print(f"Creating new FAISS index for model '{model_key}' with dimension {feature_dim}.")
77
+ index = faiss.IndexFlatL2(feature_dim)
78
+ self.faiss_indexes[model_key] = index
79
+
80
+ # Add vectors to the specific FAISS index
81
+ start_index = index.ntotal
82
+ index.add(features.astype('float32'))
83
+
84
+ # Add metadata to SQLite. The table already supports multiple models.
85
+ for i, item in enumerate(data_items):
86
+ faiss_row_index = start_index + i
87
+ self.cursor.execute(
88
+ "INSERT OR REPLACE INTO features (annotation_id, model_key, faiss_index) VALUES (?, ?, ?)",
89
+ (item.annotation.id, model_key, faiss_row_index)
90
+ )
91
+ self.conn.commit()
92
+ self.save_faiss_index(model_key) # Save the specific index that was modified
93
+
94
+ def get_features(self, data_items, model_key):
95
+ """
96
+ Retrieves features for given data items and a specific model.
97
+ """
98
+ # Get the specific index for this model
99
+ index = self._get_or_load_index(model_key)
100
+
101
+ if index is None:
102
+ # No features have ever been stored for this model
103
+ return {}, data_items
104
+
105
+ found_features = {}
106
+ not_found_items = []
107
+
108
+ ids_to_query = [item.annotation.id for item in data_items]
109
+
110
+ # Query SQLite for the given model_key
111
+ placeholders = ','.join('?' for _ in ids_to_query)
112
+ query = (f"SELECT annotation_id, faiss_index FROM features "
113
+ f"WHERE model_key=? AND annotation_id IN ({placeholders})")
114
+ params = [model_key] + ids_to_query
115
+ self.cursor.execute(query, params)
116
+
117
+ faiss_map = {ann_id: faiss_idx for ann_id, faiss_idx in self.cursor.fetchall()}
118
+
119
+ if not faiss_map:
120
+ return {}, data_items
121
+
122
+ # Reconstruct vectors from the correct FAISS index
123
+ faiss_indices = list(faiss_map.values())
124
+ retrieved_vectors = index.reconstruct_batch(faiss_indices)
125
+
126
+ id_to_vector = {ann_id: retrieved_vectors[i] for i, ann_id in enumerate(faiss_map.keys())}
127
+
128
+ for item in data_items:
129
+ ann_id = item.annotation.id
130
+ if ann_id in id_to_vector:
131
+ found_features[ann_id] = id_to_vector[ann_id]
132
+ else:
133
+ not_found_items.append(item)
134
+
135
+ return found_features, not_found_items
136
+
137
+ def get_faiss_index_to_annotation_id_map(self, model_key):
138
+ """
139
+ Retrieves a mapping from FAISS row index to annotation_id for a given model.
140
+ """
141
+ query = "SELECT faiss_index, annotation_id FROM features WHERE model_key = ?"
142
+ self.cursor.execute(query, (model_key,))
143
+ return {faiss_idx: ann_id for faiss_idx, ann_id in self.cursor.fetchall()}
144
+
145
+ def save_faiss_index(self, model_key):
146
+ """Saves a specific FAISS index to disk."""
147
+ if model_key in self.faiss_indexes:
148
+ index_to_save = self.faiss_indexes[model_key]
149
+ index_path = f"{self.index_path_base}_{model_key}.faiss"
150
+ print(f"Saving FAISS index for '{model_key}' to {index_path}")
151
+ faiss.write_index(index_to_save, index_path)
152
+
153
+ def close(self):
154
+ """Closes the database connection."""
155
+ self.conn.close()
156
+
157
+ def delete_storage(self):
158
+ """
159
+ Closes connection and deletes the DB and ALL FAISS index files.
160
+ """
161
+ self.close()
162
+
163
+ if os.path.exists(self.db_path):
164
+ try:
165
+ os.remove(self.db_path)
166
+ print(f"Deleted feature database: {self.db_path}")
167
+ except OSError as e:
168
+ print(f"Error removing database file {self.db_path}: {e}")
169
+
170
+ # Use glob to find and delete all matching index files
171
+ for index_file in glob.glob(f"{self.index_path_base}_*.faiss"):
172
+ try:
173
+ os.remove(index_file)
174
+ print(f"Deleted FAISS index: {index_file}")
175
+ except OSError as e:
176
+ print(f"Error removing index file {index_file}: {e}")
@@ -1,10 +1,10 @@
1
1
  import os
2
2
  import warnings
3
3
 
4
- from PyQt5.QtCore import Qt
4
+ from PyQt5.QtCore import Qt, pyqtSignal, pyqtSlot
5
5
  from PyQt5.QtWidgets import (QVBoxLayout, QHBoxLayout, QPushButton, QComboBox, QLabel,
6
6
  QWidget, QGroupBox, QSlider, QListWidget, QTabWidget,
7
- QLineEdit, QFileDialog, QFormLayout)
7
+ QLineEdit, QFileDialog, QFormLayout, QSpinBox)
8
8
 
9
9
  from coralnet_toolbox.MachineLearning.Community.cfg import get_available_configs
10
10
 
@@ -16,6 +16,180 @@ warnings.filterwarnings("ignore", category=DeprecationWarning)
16
16
  # ----------------------------------------------------------------------------------------------------------------------
17
17
 
18
18
 
19
+ class UncertaintySettingsWidget(QWidget):
20
+ """A widget for configuring uncertainty sampling parameters."""
21
+ parameters_changed = pyqtSignal(dict)
22
+
23
+ def __init__(self, parent=None):
24
+ super().__init__(parent)
25
+ self.setup_ui()
26
+ # Set a strong focus policy to prevent the menu from closing on interaction
27
+ self.confidence_slider.setFocusPolicy(Qt.StrongFocus)
28
+ self.margin_slider.setFocusPolicy(Qt.StrongFocus)
29
+
30
+ def setup_ui(self):
31
+ """Creates the UI controls for the parameters."""
32
+ main_layout = QFormLayout(self)
33
+ main_layout.setContentsMargins(10, 10, 10, 10)
34
+ main_layout.setSpacing(15)
35
+
36
+ # 1. Confidence Threshold
37
+ confidence_layout = QHBoxLayout()
38
+ self.confidence_slider = QSlider(Qt.Horizontal)
39
+ self.confidence_slider.setMinimum(0)
40
+ self.confidence_slider.setMaximum(100)
41
+ self.confidence_slider.setValue(60)
42
+ self.confidence_slider.setToolTip(
43
+ "Find annotations where the model's top guess\n"
44
+ "has a confidence BELOW this threshold."
45
+ )
46
+ self.confidence_label = QLabel("60%")
47
+ self.confidence_label.setMinimumWidth(40)
48
+ confidence_layout.addWidget(self.confidence_slider)
49
+ confidence_layout.addWidget(self.confidence_label)
50
+ main_layout.addRow("Max Confidence:", confidence_layout)
51
+
52
+ # 2. Margin Threshold
53
+ margin_layout = QHBoxLayout()
54
+ self.margin_slider = QSlider(Qt.Horizontal)
55
+ self.margin_slider.setMinimum(0)
56
+ self.margin_slider.setMaximum(50)
57
+ self.margin_slider.setValue(10)
58
+ self.margin_slider.setToolTip(
59
+ "Find annotations where the confidence difference\n"
60
+ "between the top two guesses is BELOW this threshold."
61
+ )
62
+ self.margin_label = QLabel("10%")
63
+ self.margin_label.setMinimumWidth(40)
64
+ margin_layout.addWidget(self.margin_slider)
65
+ margin_layout.addWidget(self.margin_label)
66
+ main_layout.addRow("Min Margin:", margin_layout)
67
+
68
+ # Connect signals
69
+ self.confidence_slider.valueChanged.connect(self._emit_parameters)
70
+ self.margin_slider.valueChanged.connect(self._emit_parameters)
71
+ self.confidence_slider.valueChanged.connect(
72
+ lambda v: self.confidence_label.setText(f"{v}%")
73
+ )
74
+ self.margin_slider.valueChanged.connect(
75
+ lambda v: self.margin_label.setText(f"{v}%")
76
+ )
77
+
78
+ @pyqtSlot()
79
+ def _emit_parameters(self):
80
+ """Gathers current values and emits them in a dictionary."""
81
+ params = self.get_parameters()
82
+ self.parameters_changed.emit(params)
83
+
84
+ def get_parameters(self):
85
+ """Returns the current parameters as a dictionary."""
86
+ return {
87
+ 'confidence': self.confidence_slider.value() / 100.0,
88
+ 'margin': self.margin_slider.value() / 100.0
89
+ }
90
+
91
+
92
+ class MislabelSettingsWidget(QWidget):
93
+ """A widget for configuring mislabel detection parameters."""
94
+ parameters_changed = pyqtSignal(dict)
95
+
96
+ def __init__(self, parent=None):
97
+ super().__init__(parent)
98
+ self.setup_ui()
99
+ # Set a default value to prevent the menu from closing on interaction
100
+ self.k_spinbox.setFocusPolicy(Qt.StrongFocus)
101
+ self.threshold_slider.setFocusPolicy(Qt.StrongFocus)
102
+
103
+ def setup_ui(self):
104
+ """Creates the UI controls for the parameters."""
105
+ main_layout = QFormLayout(self)
106
+ main_layout.setContentsMargins(10, 10, 10, 10)
107
+ main_layout.setSpacing(15)
108
+
109
+ # 1. K (Number of Neighbors)
110
+ self.k_spinbox = QSpinBox()
111
+ self.k_spinbox.setMinimum(2)
112
+ self.k_spinbox.setMaximum(50)
113
+ self.k_spinbox.setValue(5)
114
+ self.k_spinbox.setToolTip("Number of neighbors to check for each point (K).")
115
+ main_layout.addRow("Neighbors (K):", self.k_spinbox)
116
+
117
+ # 2. Agreement Threshold
118
+ threshold_layout = QHBoxLayout()
119
+ self.threshold_slider = QSlider(Qt.Horizontal)
120
+ self.threshold_slider.setMinimum(0)
121
+ self.threshold_slider.setMaximum(100)
122
+ self.threshold_slider.setValue(60)
123
+ self.threshold_slider.setToolTip(
124
+ "A point is flagged if the percentage of neighbors\n"
125
+ "with the same label is BELOW this threshold."
126
+ )
127
+
128
+ self.threshold_label = QLabel("60%")
129
+ self.threshold_label.setMinimumWidth(40)
130
+
131
+ threshold_layout.addWidget(self.threshold_slider)
132
+ threshold_layout.addWidget(self.threshold_label)
133
+ main_layout.addRow("Agreement:", threshold_layout)
134
+
135
+ # Connect signals
136
+ self.k_spinbox.valueChanged.connect(self._emit_parameters)
137
+ self.threshold_slider.valueChanged.connect(self._emit_parameters)
138
+ self.threshold_slider.valueChanged.connect(
139
+ lambda v: self.threshold_label.setText(f"{v}%")
140
+ )
141
+
142
+ @pyqtSlot()
143
+ def _emit_parameters(self):
144
+ """Gathers current values and emits them in a dictionary."""
145
+ params = self.get_parameters()
146
+ self.parameters_changed.emit(params)
147
+
148
+ def get_parameters(self):
149
+ """Returns the current parameters as a dictionary."""
150
+ return {
151
+ 'k': self.k_spinbox.value(),
152
+ 'threshold': self.threshold_slider.value() / 100.0
153
+ }
154
+
155
+
156
+ class SimilaritySettingsWidget(QWidget):
157
+ """A widget for configuring similarity search parameters (number of neighbors)."""
158
+ parameters_changed = pyqtSignal(dict)
159
+
160
+ def __init__(self, parent=None):
161
+ super().__init__(parent)
162
+ self.setup_ui()
163
+ self.k_spinbox.setFocusPolicy(Qt.StrongFocus)
164
+
165
+ def setup_ui(self):
166
+ """Creates the UI controls for the parameters."""
167
+ main_layout = QFormLayout(self)
168
+ main_layout.setContentsMargins(10, 10, 10, 10)
169
+ main_layout.setSpacing(15)
170
+
171
+ # K (Number of Neighbors)
172
+ self.k_spinbox = QSpinBox()
173
+ self.k_spinbox.setMinimum(1)
174
+ self.k_spinbox.setMaximum(200)
175
+ self.k_spinbox.setValue(10)
176
+ self.k_spinbox.setToolTip("Number of similar items to find (K).")
177
+ main_layout.addRow("Neighbors (K):", self.k_spinbox)
178
+
179
+ # Connect signals
180
+ self.k_spinbox.valueChanged.connect(self._emit_parameters)
181
+
182
+ @pyqtSlot()
183
+ def _emit_parameters(self):
184
+ params = self.get_parameters()
185
+ self.parameters_changed.emit(params)
186
+
187
+ def get_parameters(self):
188
+ return {
189
+ 'k': self.k_spinbox.value()
190
+ }
191
+
192
+
19
193
  class AnnotationSettingsWidget(QGroupBox):
20
194
  """Widget for filtering annotations by image, type, and label in a multi-column layout."""
21
195
 
@@ -220,6 +394,7 @@ class AnnotationSettingsWidget(QGroupBox):
220
394
 
221
395
  class ModelSettingsWidget(QGroupBox):
222
396
  """Widget containing model selection with tabs for different model sources."""
397
+ selection_changed = pyqtSignal()
223
398
 
224
399
  def __init__(self, main_window, parent=None):
225
400
  super(ModelSettingsWidget, self).__init__("Model Settings", parent)
@@ -243,21 +418,23 @@ class ModelSettingsWidget(QGroupBox):
243
418
  self.model_combo.addItems(["Color Features"])
244
419
  self.model_combo.insertSeparator(1) # Add a separator
245
420
 
246
- standard_models = ['yolov8n-cls.pt',
247
- 'yolov8s-cls.pt',
248
- 'yolov8m-cls.pt',
249
- 'yolov8l-cls.pt',
250
- 'yolov8x-cls.pt',
251
- 'yolo11n-cls.pt',
252
- 'yolo11s-cls.pt',
253
- 'yolo11m-cls.pt',
254
- 'yolo11l-cls.pt',
255
- 'yolo11x-cls.pt',
256
- 'yolo12n-cls.pt',
257
- 'yolo12s-cls.pt',
258
- 'yolo12m-cls.pt',
259
- 'yolo12l-cls.pt',
260
- 'yolo12x-cls.pt']
421
+ standard_models = [
422
+ 'yolov8n-cls.pt',
423
+ 'yolov8s-cls.pt',
424
+ 'yolov8m-cls.pt',
425
+ 'yolov8l-cls.pt',
426
+ 'yolov8x-cls.pt',
427
+ 'yolo11n-cls.pt',
428
+ 'yolo11s-cls.pt',
429
+ 'yolo11m-cls.pt',
430
+ 'yolo11l-cls.pt',
431
+ 'yolo11x-cls.pt',
432
+ 'yolo12n-cls.pt',
433
+ 'yolo12s-cls.pt',
434
+ 'yolo12m-cls.pt',
435
+ 'yolo12l-cls.pt',
436
+ 'yolo12x-cls.pt'
437
+ ]
261
438
 
262
439
  self.model_combo.addItems(standard_models)
263
440
 
@@ -267,9 +444,6 @@ class ModelSettingsWidget(QGroupBox):
267
444
  self.model_combo.addItems(list(community_configs.keys()))
268
445
 
269
446
  self.model_combo.setCurrentText('Color Features')
270
- # Connect selection change to update feature mode field state
271
- self.model_combo.currentTextChanged.connect(self._update_feature_mode_state)
272
-
273
447
  model_select_layout.addRow("Model:", self.model_combo)
274
448
 
275
449
  self.tabs.addTab(model_select_tab, "Select Model")
@@ -293,18 +467,21 @@ class ModelSettingsWidget(QGroupBox):
293
467
 
294
468
  main_layout.addWidget(self.tabs)
295
469
 
296
- # Connect tab change to update feature mode state
297
- self.tabs.currentChanged.connect(self._update_feature_mode_state)
470
+ # Connect all relevant widgets to a single slot that emits the new signal
471
+ self.model_combo.currentTextChanged.connect(self._on_selection_changed)
472
+ self.tabs.currentChanged.connect(self._on_selection_changed)
473
+ self.model_path_edit.textChanged.connect(self._on_selection_changed)
298
474
 
299
475
  # Add feature extraction mode selection outside of tabs
300
476
  feature_mode_layout = QFormLayout()
301
477
  self.feature_mode_combo = QComboBox()
302
478
  self.feature_mode_combo.addItems(["Predictions", "Embed Features"])
479
+ self.feature_mode_combo.currentTextChanged.connect(self._on_selection_changed)
303
480
  feature_mode_layout.addRow("Feature Mode:", self.feature_mode_combo)
304
481
  main_layout.addLayout(feature_mode_layout)
305
482
 
306
- # Initialize the feature mode state based on current selection
307
- self._update_feature_mode_state()
483
+ # Initialize the feature mode state and emit the first signal
484
+ self._on_selection_changed()
308
485
 
309
486
  def browse_for_model(self):
310
487
  """Open a file dialog to browse for model files."""
@@ -318,6 +495,12 @@ class ModelSettingsWidget(QGroupBox):
318
495
  )
319
496
  if file_path:
320
497
  self.model_path_edit.setText(file_path)
498
+
499
+ @pyqtSlot()
500
+ def _on_selection_changed(self):
501
+ """Central slot to handle any change in model selection and emit a single signal."""
502
+ self._update_feature_mode_state()
503
+ self.selection_changed.emit()
321
504
 
322
505
  def _update_feature_mode_state(self, *args):
323
506
  """Update the enabled state of the feature mode field based on the current model selection."""
@@ -26,18 +26,29 @@ class GlobalEventFilter(QObject):
26
26
  self.auto_distill_deploy_model_dialog = main_window.auto_distill_deploy_model_dialog
27
27
 
28
28
  def eventFilter(self, obj, event):
29
+ # Check for explorer window first - this applies to all event types
30
+ if hasattr(self.main_window, 'explorer_window') and self.main_window.explorer_window:
31
+ # Special exception for WASD keys which should always work
32
+ if event.type() == QEvent.KeyPress and event.key() in [Qt.Key_W, Qt.Key_A, Qt.Key_S, Qt.Key_D] and \
33
+ event.modifiers() & Qt.ControlModifier:
34
+ self.label_window.handle_wasd_key(event.key())
35
+ return True
36
+
37
+ # For all other events when explorer is visible, pass them through
38
+ return False
39
+
40
+ # Now handle keyboard events
29
41
  if event.type() == QEvent.KeyPress:
30
42
  if event.modifiers() & Qt.ControlModifier and not (event.modifiers() & Qt.ShiftModifier):
31
-
32
- # Handle Tab key for switching between Select and Annotation tools
33
- if event.key() == Qt.Key_Alt:
34
- self.main_window.switch_back_to_tool()
35
- return True
36
-
37
43
  # Handle WASD keys for selecting Label
38
44
  if event.key() in [Qt.Key_W, Qt.Key_A, Qt.Key_S, Qt.Key_D]:
39
45
  self.label_window.handle_wasd_key(event.key())
40
46
  return True
47
+
48
+ # Handle Alt key for switching between Select and Annotation tools
49
+ if event.key() == Qt.Key_Alt:
50
+ self.main_window.switch_back_to_tool()
51
+ return True
41
52
 
42
53
  # Handle hotkey for image classification prediction
43
54
  if event.key() == Qt.Key_1:
@@ -74,22 +85,12 @@ class GlobalEventFilter(QObject):
74
85
 
75
86
  # Delete (backspace or delete key) selected annotations when select tool is active
76
87
  if event.key() == Qt.Key_Delete or event.key() == Qt.Key_Backspace:
77
-
78
- # First, check if the Explorer window exists and is the active window
79
- if (self.main_window.explorer_window and
80
- self.main_window.explorer_window.isActiveWindow()):
81
-
82
- # If it is, let the Explorer handle its own key press events.
83
- # Returning False passes the event along instead of consuming it.
84
- return False
85
-
86
- # If Explorer is not active, proceed with the original logic for the main window
87
88
  if self.main_window.select_tool_action.isChecked():
88
89
  if self.annotation_window.selected_annotations:
89
90
  self.annotation_window.delete_selected_annotations()
90
91
  # Consume the event so it doesn't do anything else
91
92
  return True
92
-
93
+
93
94
  # Handle image cycling hotkeys
94
95
  if event.key() == Qt.Key_Up and event.modifiers() == (Qt.AltModifier):
95
96
  self.image_window.cycle_previous_image()
@@ -1616,6 +1616,49 @@ class MainWindow(QMainWindow):
1616
1616
  except Exception as e:
1617
1617
  QMessageBox.critical(self, "Critical Error", f"{e}")
1618
1618
 
1619
+ def set_main_window_enabled_state(self, enable_list=None, disable_list=None):
1620
+ """
1621
+ Modular method to enable/disable widgets and actions in the main window.
1622
+ - enable_list: list of widgets/actions to enable
1623
+ - disable_list: list of widgets/actions to disable
1624
+ If both are None, enables everything.
1625
+ """
1626
+ # All main widgets/actions to consider
1627
+ all_widgets = [
1628
+ self.toolbar,
1629
+ self.menu_bar,
1630
+ self.image_window,
1631
+ self.label_window,
1632
+ self.confidence_window,
1633
+ self.annotation_window,
1634
+ # Status bar widgets
1635
+ *(self.status_bar_layout.itemAt(i).widget() for i in range(self.status_bar_layout.count()))
1636
+ ]
1637
+ # Remove None entries (in case any status bar slot is empty)
1638
+ all_widgets = [w for w in all_widgets if w is not None]
1639
+
1640
+ # If neither list is provided, enable everything
1641
+ if enable_list is None and disable_list is None:
1642
+ for w in all_widgets:
1643
+ w.setEnabled(True)
1644
+ return
1645
+
1646
+ # Disable everything by default
1647
+ for w in all_widgets:
1648
+ w.setEnabled(False)
1649
+
1650
+ # Enable specified widgets/actions
1651
+ if enable_list:
1652
+ for w in enable_list:
1653
+ if w is not None:
1654
+ w.setEnabled(True)
1655
+
1656
+ # Disable specified widgets/actions (overrides enable if both present)
1657
+ if disable_list:
1658
+ for w in disable_list:
1659
+ if w is not None:
1660
+ w.setEnabled(False)
1661
+
1619
1662
  def open_explorer_window(self):
1620
1663
  """Open the Explorer window, moving the LabelWindow into it."""
1621
1664
  # Check if there are any images in the project
@@ -1638,24 +1681,33 @@ class MainWindow(QMainWindow):
1638
1681
  self.explorer_window = ExplorerWindow(self)
1639
1682
 
1640
1683
  # Move the label_window from the main layout to the explorer
1641
- # The ExplorerWindow's __init__ will handle adding it to its own layout.
1642
1684
  self.left_layout.removeWidget(self.label_window)
1643
1685
  self.label_window.setParent(self.explorer_window.left_panel) # Re-parent
1644
1686
  self.explorer_window.left_layout.insertWidget(1, self.label_window) # Add to explorer layout
1645
1687
 
1646
- # Make the explorer window modal to block interaction with main window
1647
- self.explorer_window.setWindowModality(Qt.ApplicationModal)
1648
- # Disable the main window explicitly
1649
- self.setEnabled(False)
1688
+ # Disable all main window widgets except select few
1689
+ self.set_main_window_enabled_state(
1690
+ enable_list=[self.annotation_window,
1691
+ self.label_window,
1692
+ self.transparency_widget],
1693
+ disable_list=[self.toolbar,
1694
+ self.menu_bar,
1695
+ self.image_window,
1696
+ self.confidence_window]
1697
+ )
1650
1698
 
1651
1699
  self.explorer_window.showMaximized()
1652
1700
  self.explorer_window.activateWindow()
1653
1701
  self.explorer_window.raise_()
1702
+
1654
1703
  except Exception as e:
1655
1704
  QMessageBox.critical(self, "Critical Error", f"{e}")
1656
1705
  if self.explorer_window:
1657
1706
  self.explorer_window.close() # Ensure cleanup
1707
+
1658
1708
  self.explorer_window = None
1709
+ # Re-enable everything if there was an error
1710
+ self.set_main_window_enabled_state()
1659
1711
 
1660
1712
  def explorer_closed(self):
1661
1713
  """Handle the explorer window being closed."""
@@ -1667,8 +1719,8 @@ class MainWindow(QMainWindow):
1667
1719
  self.label_window.resizeEvent(None)
1668
1720
  self.resizeEvent(None)
1669
1721
 
1670
- # Re-enable the main window
1671
- self.setEnabled(True)
1722
+ # Re-enable all main window widgets
1723
+ self.set_main_window_enabled_state()
1672
1724
 
1673
1725
  # Clean up reference
1674
1726
  self.explorer_window = None
@@ -1,6 +1,6 @@
1
1
  """Top-level package for CoralNet-Toolbox."""
2
2
 
3
- __version__ = "0.0.68"
3
+ __version__ = "0.0.69"
4
4
  __author__ = "Jordan Pierce"
5
5
  __email__ = "jordan.pierce@noaa.gov"
6
6
  __credits__ = "National Center for Coastal and Ocean Sciences (NCCOS)"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: coralnet-toolbox
3
- Version: 0.0.68
3
+ Version: 0.0.69
4
4
  Summary: Tools for annotating and developing ML models for benthic imagery
5
5
  Author-email: Jordan Pierce <jordan.pierce@noaa.gov>
6
6
  License: MIT License
@@ -22,6 +22,7 @@ Requires-Dist: open-clip-torch>=2.20.0
22
22
  Requires-Dist: supervision>=0.24.0
23
23
  Requires-Dist: scikit-learn
24
24
  Requires-Dist: umap-learn
25
+ Requires-Dist: faiss-cpu
25
26
  Requires-Dist: pycocotools
26
27
  Requires-Dist: ujson
27
28
  Requires-Dist: timm==0.9.2
@@ -171,11 +172,12 @@ Enhance your CoralNet experience with these tools:
171
172
  - Uses `Grounding DINO`, `OWLViT`, `OmDetTurbo`
172
173
  - 📻 Tune: Tune hyperparameters to identify ideal training conditions
173
174
  - 🧠 Train: Build local patch-based classifiers, object detection, and instance segmentation models
174
- - 🔮 Deploy: Use trained models for predictions
175
+ - 🧙‍♂️ Deploy: Use trained models for predictions
175
176
  - 📊 Evaluation: Evaluate model performance
176
177
  - 🚀 Optimize: Productionize models for faster inferencing
177
178
  - ⚙️ Batch Inference: Perform predictions on multiple images, automatically
178
179
  - 🎞️ Video Inference: Perform predictions on a video in real-time, record the output and analytics
180
+ - 🔮 Explorer: Cluster, view, and re-label annotations using embeddings, mapped from feature-space
179
181
  - ↔️ I/O: Import and Export annotations from / to CoralNet, Viscore, and TagLab
180
182
  - Export annotations as [GeoJSONs](https://datatracker.ietf.org/doc/html/rfc7946), segmentation masks
181
183
  - 📸 YOLO: Import and Export YOLO datasets for machine learning