添加命令行传入参数支持
This commit is contained in:
parent
bd0d65d4ea
commit
a6e1fbf37e
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user