accusleepy 0.1.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

accusleepy/config.json

@@ -18,5 +18,6 @@
             "is_scored": true,
             "frequency": 0.55
         }
-    ]
-}
+    ],
+    "default_epoch_length": 2.5
+}

accusleepy/constants.py

@@ -35,3 +35,5 @@ LABEL_COL = "label"
 # recording list file header:
 RECORDING_LIST_NAME = "recording_list"
 RECORDING_LIST_FILE_TYPE = ".json"
+# key for default epoch length in config
+DEFAULT_EPOCH_LENGTH_KEY = "default_epoch_length"

accusleepy/fileio.py

@@ -11,6 +11,7 @@ from accusleepy.brain_state_set import BRAIN_STATES_KEY, BrainState, BrainStateS
 from accusleepy.constants import (
     BRAIN_STATE_COL,
     CONFIG_FILE,
+    DEFAULT_EPOCH_LENGTH_KEY,
     EEG_COL,
     EMG_COL,
     MIXTURE_MEAN_COL,
@@ -82,7 +83,9 @@ def load_model(filename: str) -> tuple[SSANN, int | float, int, str, dict]:
         (default or real-time), set of brain state options
         used when training the model
     """
-    state_dict = torch.load(filename, weights_only=True)
+    state_dict = torch.load(
+        filename, weights_only=True, map_location=torch.device("cpu")
+    )
     epoch_length = state_dict.pop("epoch_length")
     epochs_per_img = state_dict.pop("epochs_per_img")
     model_type = state_dict.pop("model_type")
@@ -141,10 +144,10 @@ def save_labels(labels: np.array, filename: str) -> None:
     pd.DataFrame({BRAIN_STATE_COL: labels}).to_csv(filename, index=False)
 
 
-def load_config() -> BrainStateSet:
+def load_config() -> tuple[BrainStateSet, int | float]:
     """Load configuration file with brain state options
 
-    :return: set of brain state options
+    :return: set of brain state options and default epoch length
     """
     with open(
         os.path.join(os.path.dirname(os.path.abspath(__file__)), CONFIG_FILE), "r"
@@ -152,18 +155,23 @@ def load_config() -> BrainStateSet:
         data = json.load(f)
     return BrainStateSet(
         [BrainState(**b) for b in data[BRAIN_STATES_KEY]], UNDEFINED_LABEL
-    )
+    ), data[DEFAULT_EPOCH_LENGTH_KEY]
 
 
-def save_config(brain_state_set: BrainStateSet) -> None:
+def save_config(
+    brain_state_set: BrainStateSet, default_epoch_length: int | float
+) -> None:
     """Save configuration of brain state options to json file
 
     :param brain_state_set: set of brain state options
+    :param default_epoch_length: epoch length to use when the GUI starts
     """
+    output_dict = brain_state_set.to_output_dict()
+    output_dict.update({DEFAULT_EPOCH_LENGTH_KEY: default_epoch_length})
     with open(
         os.path.join(os.path.dirname(os.path.abspath(__file__)), CONFIG_FILE), "w"
     ) as f:
-        json.dump(brain_state_set.to_output_dict(), f, indent=4)
+        json.dump(output_dict, f, indent=4)
 
 
 def load_recording_list(filename: str) -> list[Recording]:

accusleepy/gui/main.py

@@ -40,6 +40,7 @@ from accusleepy.fileio import (
     save_model,
     save_recording_list,
 )
+from accusleepy.gui.text.main_guide_text import MAIN_GUIDE_TEXT
 from accusleepy.gui.manual_scoring import ManualScoringWindow
 from accusleepy.gui.primary_window import Ui_PrimaryWindow
 from accusleepy.signal_processing import (
@@ -52,8 +53,7 @@ from accusleepy.signal_processing import (
 # max number of messages to display
 MESSAGE_BOX_MAX_DEPTH = 50
 LABEL_LENGTH_ERROR = "label file length does not match recording length"
-# relative path to user manual txt file
-USER_MANUAL_FILE = "text/main_guide.txt"
+# relative path to config guide txt file
 CONFIG_GUIDE_FILE = "text/config_guide.txt"
 
 
@@ -80,12 +80,12 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
         self.setWindowTitle("AccuSleePy")
 
         # fill in settings tab
-        self.brain_state_set = load_config()
+        self.brain_state_set, self.epoch_length = load_config()
         self.settings_widgets = None
         self.initialize_settings_tab()
 
         # initialize info about the recordings, classification data / settings
-        self.epoch_length = 0
+        self.ui.epoch_length_input.setValue(self.epoch_length)
         self.model = None
         self.only_overwrite_undefined = False
         self.min_bout_length = 5
@@ -267,7 +267,9 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
             )
             return
         if self.training_image_dir == "":
-            self.show_message("ERROR: no folder selected for training images.")
+            self.show_message(
+                ("ERROR: no output location selected for training images.")
+            )
             return
 
         # check some inputs for each recording
@@ -288,23 +290,28 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
         if not model_filename:
             self.show_message("Model training canceled, no filename given")
 
-        # create image folder
-        if os.path.exists(self.training_image_dir):
+        # create (probably temporary) image folder
+        temp_image_dir = os.path.join(
+            self.training_image_dir,
+            "images_" + datetime.datetime.now().strftime("%Y%m%d%H%M"),
+        )
+
+        if os.path.exists(temp_image_dir):  # unlikely
             self.show_message(
                 "Warning: training image folder exists, will be overwritten"
             )
-        os.makedirs(self.training_image_dir, exist_ok=True)
+        os.makedirs(temp_image_dir, exist_ok=True)
 
         # create training images
         self.show_message(
-            (f"Creating training images in {self.training_image_dir}, please wait...")
+            (f"Creating training images in {temp_image_dir}, please wait...")
         )
         self.ui.message_area.repaint()
         QtWidgets.QApplication.processEvents()
         print("Creating training images")
         failed_recordings = create_training_images(
             recordings=self.recordings,
-            output_path=self.training_image_dir,
+            output_path=temp_image_dir,
             epoch_length=self.epoch_length,
             epochs_per_img=self.training_epochs_per_img,
             brain_state_set=self.brain_state_set,
@@ -328,10 +335,8 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
         QtWidgets.QApplication.processEvents()
         print("Training model")
         model = train_model(
-            annotations_file=os.path.join(
-                self.training_image_dir, ANNOTATIONS_FILENAME
-            ),
-            img_dir=self.training_image_dir,
+            annotations_file=os.path.join(temp_image_dir, ANNOTATIONS_FILENAME),
+            img_dir=temp_image_dir,
             mixture_weights=self.brain_state_set.mixture_weights,
             n_classes=self.brain_state_set.n_classes,
         )
@@ -348,20 +353,18 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
 
         # optionally delete images
         if self.delete_training_images:
-            shutil.rmtree(self.training_image_dir)
+            shutil.rmtree(temp_image_dir)
 
         self.show_message(f"Training complete, saved model to {model_filename}")
 
-    def set_training_folder(self):
+    def set_training_folder(self) -> None:
+        """Select location in which to create a folder for training images"""
         training_folder_parent = QtWidgets.QFileDialog.getExistingDirectory(
             self, "Select directory for training images"
         )
         if training_folder_parent:
-            self.training_image_dir = os.path.join(
-                training_folder_parent,
-                "images_" + datetime.datetime.now().strftime("%Y%m%d%H%M"),
-            )
-            self.ui.image_folder_label.setText(self.training_image_dir)
+            self.training_image_dir = training_folder_parent
+            self.ui.image_folder_label.setText(training_folder_parent)
 
     def update_image_deletion(self) -> None:
         """Update choice of whether to delete images after training"""
@@ -1002,15 +1005,8 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
 
     def show_user_manual(self) -> None:
         """Show a popup window with the user manual"""
-        user_manual_file = open(
-            os.path.join(os.path.dirname(os.path.abspath(__file__)), USER_MANUAL_FILE),
-            "r",
-        )
-        user_manual_text = user_manual_file.read()
-        user_manual_file.close()
-
         label_widget = QtWidgets.QLabel()
-        label_widget.setText(user_manual_text)
+        label_widget.setText(MAIN_GUIDE_TEXT)
         scroll_area = QtWidgets.QScrollArea()
         scroll_area.setStyleSheet("background-color: white;")
         scroll_area.setWidget(label_widget)
@@ -1108,6 +1104,7 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
         }
 
         # update widget state to display current config
+        self.ui.default_epoch_input.setValue(self.epoch_length)
         states = {b.digit: b for b in self.brain_state_set.brain_states}
         for digit in range(10):
             if digit in states.keys():
@@ -1245,7 +1242,7 @@ class AccuSleepWindow(QtWidgets.QMainWindow):
         self.brain_state_set = BrainStateSet(brain_states, UNDEFINED_LABEL)
 
         # save to file
-        save_config(self.brain_state_set)
+        save_config(self.brain_state_set, self.ui.default_epoch_input.value())
         self.ui.save_config_status.setText("configuration saved")
 
 

accusleepy/gui/manual_scoring.py

@@ -111,7 +111,7 @@ class ManualScoringWindow(QtWidgets.QDialog):
         self.setWindowTitle("AccuSleePy manual scoring window")
 
         # load set of valid brain states
-        self.brain_state_set = load_config()
+        self.brain_state_set, _ = load_config()
 
         # initial setting for number of epochs to show in the lower plot
         self.epochs_to_show = 5
@@ -833,7 +833,7 @@ class ManualScoringWindow(QtWidgets.QDialog):
             self.adjust_upper_figure_x_limits()
 
         # update parts of lower plot
-        old_window_center = round(self.epochs_to_show / 2) + self.lower_left_epoch
+        old_window_center = round((self.epochs_to_show - 1) / 2) + self.lower_left_epoch
         # change the window bounds if needed
         if self.epoch < old_window_center and self.lower_left_epoch > 0:
             self.lower_left_epoch -= 1