添加命令行传入参数支持

This commit is contained in:
marques 2022-04-20 22:17:22 +08:00
parent bd0d65d4ea
commit a6e1fbf37e
2 changed files with 21 additions and 5 deletions

View File

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

View File

@ -67,4 +67,9 @@ Quality_Relabel 主目录
2022-04-15 2206
1. 删除了PSG开头部分不存在于心晓中的标签以后不用计算shifting
1. ~~删除了PSG开头部分不存在于心晓中的标签以后不用计算shifting~~
2022-04-17 2000
1. 添加命令行传入参数功能