-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
53 lines (45 loc) · 2.36 KB
/
main.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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# Created by Bastien Rigaud at 17/03/2022
# Bastien Rigaud, PhD
# Laboratoire Traitement du Signal et de l'Image (LTSI), INSERM U1099
# Campus de Beaulieu, Université de Rennes 1
# 35042 Rennes, FRANCE
# bastien.rigaud@univ-rennes1.fr
# Description:
from PlotNIfTI import PlotNifti
def main():
# image_path = r"C:\Data\Data_test\plot_cervix_ct\image.nii.gz"
# segmentation_paths = [r"C:\Data\Data_test\plot_cervix_ct\CTVT.nii.gz",
# r"C:\Data\Data_test\plot_cervix_ct\Bladder.nii.gz",
# r"C:\Data\Data_test\plot_cervix_ct\Rectum.nii.gz",
# r"C:\Data\Data_test\plot_cervix_ct\Sigmoid.nii.gz",
# r"C:\Data\Data_test\plot_cervix_ct\BowelBag.nii.gz"]
#
# views = ['sagittal', 'axial', 'coronal']
# # segmentation_names is optional, but useful to add a colormap
# segmentation_names = ['CTVT', 'Bladder', 'Rectum', 'Sigmoid', 'BowelBag']
# plot_object = PlotNifti(image_path=image_path, segmentation_paths=segmentation_paths,
# show_contour=True, show_filled=True, transparency=0.20, get_at_centroid=True,
# segmentation_names=segmentation_names)
#
# for view in views:
# output_path = r"example\screenshot_{}.png".format(view)
# plot_object.set_view(view)
# plot_object.set_output_path(output_path)
# plot_object.generate_plot()
image_path = r"C:\Data\Data_test\plot\image.nii.gz"
segmentation_paths = [r"C:\Data\Data_test\plot\Prostate_DL_v0.nii.gz",
r"C:\Data\Data_test\plot\Bladder_DL_v0.nii.gz",
r"C:\Data\Data_test\plot\Rectum_DL_v0.nii.gz"]
views = ['sagittal']
# segmentation_names is optional, but useful to add a colormap
segmentation_names = ['Prostate', 'Bladder', 'Rectum']
plot_object = PlotNifti(image_path=image_path, segmentation_paths=segmentation_paths,
show_contour=True, show_filled=True, transparency=0.20, get_at_centroid=True,
segmentation_names=segmentation_names)
for view in views:
output_path = r"C:\Data\Data_test\plot\screenshot_{}.png".format(view)
plot_object.set_view(view)
plot_object.set_output_path(output_path)
plot_object.generate_plot()
if __name__ == '__main__':
main()