Skip to content

Commit

Permalink
Add code to save 3D plot showing lung mask with cli argument --plot3d.
Browse files Browse the repository at this point in the history
  • Loading branch information
sbrodehl committed Aug 24, 2020
1 parent 4e2d847 commit afed833
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions luvox.py
Original file line number Diff line number Diff line change
Expand Up @@ -434,6 +434,12 @@ def run_on_case(case, logstr, outfile, prefix, seriesiuid, study_uid, suid, tbz_
ov
)

# create a 3D for this series
if kwargs.get("plot3d", False):
plotting.plot_3d(lung_mask, desc="LungMask",
StudyInstanceUID=img.StudyInstanceUID, SeriesInstanceUID=img.SeriesInstanceUID)

# end plot time
plot_time = time.time() - plot_start_time
result = "{},{},{},{},{},{},{},{},{},{}".format(
suid, img.SeriesInstanceUID,
Expand Down

0 comments on commit afed833

Please sign in to comment.