-
Notifications
You must be signed in to change notification settings - Fork 1
/
SegmentujObrazyJPET.py
34 lines (26 loc) · 1.05 KB
/
SegmentujObrazyJPET.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
"""
Main script. Supports UI.
Główny plik projektu. Obsługuje komunikację z użytkownikiem
"""
from lib import *
from segmentVolume import SegmentVolume
import argparse
import json
# Define parser
parser = argparse.ArgumentParser(description=TEKST_POWITALNY, formatter_class=argparse.RawDescriptionHelpFormatter)
# Positional arguments - algorithm i datapath
parser.add_argument('alg', metavar='ALGORITHM', type=str, help=INFO_ALG)
parser.add_argument('dataPath', metavar='PATH_TO_DATA', type=str, help=INFO_DATA)
# Optional arguments
parser.add_argument('--savePickle', action='store_true', help=INFO_SAVE_PICKLE)
parser.add_argument('--saveSlices', action='store_true', help=INFO_SAVE_SLICES)
parser.add_argument('--algParams', metavar=' KEYWORD_PARAMS_TO_ALG', type=json.loads) # json/dict arguments
# Main
args = parser.parse_args()
# print('Podano:', args)
segObj = SegmentVolume(args.dataPath)
segObj.segmentation(args.alg, args.algParams)
if args.savePickle:
segObj.saveVolumeAsPickle()
if args.saveSlices:
segObj.saveSlicesAsPng()