Diff of /data_preproc.py [000000] .. [eaa663]

Switch to side-by-side view

--- a
+++ b/data_preproc.py
@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Sun Apr 21 13:32:56 2019
+
+@author: Winham
+
+data_preproc.py:用于人工标记后的文件整理
+注意:由于下列代码中包含了对文件的删除,因此在原始人工标记后的文件中
+仅能运行一次。建议运行前先将原始文件备份!!!若遇到错误可重新恢复并
+重新执行。运行前先在同目录下新建一个文件夹119_SEG
+
+"""
+
+import os
+import numpy as np
+import scipy.io as sio
+
+path = 'G:/ECG_UNet/119_MASK/'               # 原始文件目录
+seg_path = 'G:/ECG_UNet/119_SEG/'            # 存储信号.npy文件
+
+files = os.listdir(path)
+
+for i in range(len(files)):
+    file_name = files[i]
+    print(file_name + ' ' + str(i+1))
+    if file_name.endswith('.json'):          # 只取已经人工标记好的信号段,即有.json文件配套
+        name = file_name[:-5]
+        mat_name = name + '.mat'
+        sig = sio.loadmat(path+mat_name)['seg_t'].squeeze()
+        np.save(seg_path+name+'.npy', sig)
+    elif file_name.startswith('ann') or file_name.endswith('.png'):
+        os.remove(path+file_name)
+
+rest_files = os.listdir(path)
+for i in range(len(rest_files)):
+    if rest_files[i].endswith('.mat'):
+        os.remove(path+rest_files[i])