diff --git a/Main_Quality_Relabel.py b/Main_Quality_Relabel.py index 09d420f..5475c5b 100644 --- a/Main_Quality_Relabel.py +++ b/Main_Quality_Relabel.py @@ -7,6 +7,7 @@ @email:2021022362@m.scnu.edu.cn @time:2022/03/28 """ +import argparse from pathlib import Path from utils.Quality_Relabel import Quality_Relabel @@ -40,9 +41,19 @@ sampNo = 670 start_bcg_index = 0 if __name__ == '__main__': - qualityRelabel = Quality_Relabel(all_path=all_path, sampNo=sampNo, frequency=frequency, bcg_frequency=bcg_frequency, + parser = argparse.ArgumentParser() + parser.add_argument('-n', '--sampNo', nargs='?', type=int, default=sampNo, help='samp number') + parser.add_argument('-i', '--start_bcg_index', nargs='?', type=int, default=start_bcg_index, help='samp number') + parser.add_argument('-f', '--front_add_second', nargs='?', type=int, default=front_add_second, + help='Add x second signal in front of the event') + parser.add_argument('-b', '--back_add_second', nargs='?', type=int, default=back_add_second, + help='Add x second signal after the event') + opt = parser.parse_args() + + qualityRelabel = Quality_Relabel(all_path=all_path, sampNo=opt.sampNo, frequency=frequency, + bcg_frequency=bcg_frequency, focus_event_list=focus_event_list) - qualityRelabel.show_all_event(start_bcg_index=start_bcg_index, - front_add_second=front_add_second, - back_add_second=back_add_second) + qualityRelabel.show_all_event(start_bcg_index=opt.start_bcg_index, + front_add_second=opt.front_add_second, + back_add_second=opt.back_add_second) diff --git a/README.md b/README.md index 2e1a1ae..57cd032 100644 --- a/README.md +++ b/README.md @@ -67,4 +67,9 @@ Quality_Relabel 主目录 2022-04-15 22:06 -​ 1. 删除了PSG开头部分不存在于心晓中的标签,以后不用计算shifting + 1. ~~删除了PSG开头部分不存在于心晓中的标签,以后不用计算shifting~~ + +2022-04-17 20:00 + +​ 1. 添加命令行传入参数功能 +