signal_label/bcg_quality_label
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复了一处致命错误

Browse Source
This commit is contained in:
Yorusora 2025-02-14 16:43:35 +08:00
parent 0f16a7707f
commit a3126781e9
2 changed files with 15 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -4,7 +4,7 @@
项目作者信息
@author:Yosa
@email:2023025086@m.scnu.edu.cn
@time:2025/2/10
@time:2025/2/11
```
本程序提供符合用户直接的操作界面，用于对BCG信号片段的质量进行打标。

40
bcg_quality_label.py
View File

@ -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)