添加命令行传入参数支持
This commit is contained in:
parent
bd0d65d4ea
commit
a6e1fbf37e
@ -7,6 +7,7 @@
|
|||||||
@email:2021022362@m.scnu.edu.cn
|
@email:2021022362@m.scnu.edu.cn
|
||||||
@time:2022/03/28
|
@time:2022/03/28
|
||||||
"""
|
"""
|
||||||
|
import argparse
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from utils.Quality_Relabel import Quality_Relabel
|
from utils.Quality_Relabel import Quality_Relabel
|
||||||
|
|
||||||
@ -40,9 +41,19 @@ sampNo = 670
|
|||||||
start_bcg_index = 0
|
start_bcg_index = 0
|
||||||
|
|
||||||
if __name__ == '__main__':
|
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)
|
focus_event_list=focus_event_list)
|
||||||
|
|
||||||
qualityRelabel.show_all_event(start_bcg_index=start_bcg_index,
|
qualityRelabel.show_all_event(start_bcg_index=opt.start_bcg_index,
|
||||||
front_add_second=front_add_second,
|
front_add_second=opt.front_add_second,
|
||||||
back_add_second=back_add_second)
|
back_add_second=opt.back_add_second)
|
||||||
|
Loading…
Reference in New Issue
Block a user