diff --git a/README.md b/README.md index 637a2ad..2196f00 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,7 @@ 项目作者信息 @author:Yosa @email:2023025086@m.scnu.edu.cn -@time:2025/2/10 +@time:2025/2/11 ``` 本程序提供符合用户直接的操作界面,用于对BCG信号片段的质量进行打标。 diff --git a/bcg_quality_label.py b/bcg_quality_label.py index 17a5303..d58455d 100644 --- a/bcg_quality_label.py +++ b/bcg_quality_label.py @@ -307,11 +307,11 @@ class MainWindow(QMainWindow, Ui_MainWindow): if self.radioButton_10s_mode.isChecked(): self.data_part_num = len(self.data) // 10000 self.groupBox_labelDisplay_10s.setVisible(True) - self.lineEdit_savepath.setText(str(self.data_and_artifact_Path / "SQ_label_10s.csv")) + self.lineEdit_savepath.setText(str(Path(self.lineEdit_savepath.text()) / "SQ_label_10s.csv")) elif self.radioButton_30s_mode.isChecked(): self.data_part_num = len(self.data) // 30000 self.groupBox_labelDisplay_30s.setVisible(True) - self.lineEdit_savepath.setText(str(self.data_and_artifact_Path / "SQ_label_30s.csv")) + self.lineEdit_savepath.setText(str(Path(self.lineEdit_savepath.text()) / "SQ_label_30s.csv")) if self.data_part_num == 0: info("Failed to Input Data!") self.textBrowser_update("操作:导入数据失败,请检查数据长度") @@ -367,32 +367,20 @@ class MainWindow(QMainWindow, Ui_MainWindow): # 尝试获取之前打标的存档,若不存在存档,则创建新存档 if self.radioButton_10s_mode.isChecked() == True: self.pushButton_Ctype_signal_label.setEnabled(True) - if not Path(self.data_and_artifact_Path / "SQ_label_10s.csv").exists(): - self.label = np.full(self.data_part_num, "f") - self.df_label = DataFrame(columns=["label", "remark"]) - self.df_label["label"] = self.label - self.df_label["remark"] = "" - self.df_label["label"] = self.df_label["label"].astype(str) - self.df_label["remark"] = self.df_label["remark"].astype(str) - else: - self.df_label = read_csv(self.data_and_artifact_Path / "SQ_label_10s.csv", encoding="gbk") - self.label = self.df_label["label"].astype(str) - self.df_label.to_csv(self.data_and_artifact_Path / "SQ_label_10s.csv", index=False, encoding="gbk") - self.lineEdit_savepath.setText(str(self.data_and_artifact_Path / "SQ_label_10s.csv")) elif self.radioButton_30s_mode.isChecked() == True: self.pushButton_Ctype_signal_label.setEnabled(False) - if not Path(self.data_and_artifact_Path / "SQ_label_30s.csv").exists(): - self.label = np.full(self.data_part_num, "f") - self.df_label = DataFrame(columns=["label", "remark"]) - self.df_label["label"] = self.label - self.df_label["remark"] = "" - self.df_label["label"] = self.df_label["label"].astype(str) - self.df_label["remark"] = self.df_label["remark"].astype(str) - else: - self.df_label = read_csv(self.data_and_artifact_Path / "SQ_label_30s.csv", encoding="gbk") - self.label = self.df_label["label"].astype(str) - self.df_label.to_csv(self.data_and_artifact_Path / "SQ_label_30s.csv", index=False, encoding="gbk") - self.lineEdit_savepath.setText(str(self.data_and_artifact_Path / "SQ_label_30s.csv")) + if not Path(self.lineEdit_savepath.text()).exists(): + self.label = np.full(self.data_part_num, "f") + self.df_label = DataFrame(columns=["label", "remark"]) + self.df_label["label"] = self.label + self.df_label["remark"] = "" + self.df_label["label"] = self.df_label["label"].astype(str) + self.df_label["remark"] = self.df_label["remark"].astype(str) + else: + self.df_label = read_csv(self.lineEdit_savepath.text(), encoding="gbk") + self.label = self.df_label["label"].astype(str) + self.df_label.to_csv(self.lineEdit_savepath.text(), index=False, encoding="gbk") + self.lineEdit_savepath.setText(str(self.lineEdit_savepath.text())) # 界面状态更新 self.groupBox_inputSetting.setEnabled(False)