Commit 4e080d8a authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Add an output option to save the figure instead of showing it in case of using...

Add an output option to save the figure instead of showing it in case of using this command with ssh
parent 98898409
Pipeline #17626 passed with stage
in 30 minutes and 44 seconds
......@@ -40,6 +40,7 @@ def command_line_arguments(command_line_parameters):
parser.add_argument('-F', '--font-size', type=int, default=16, help = 'Select the font size for the annotation names')
parser.add_argument('-C', '--font-color', default = 'b', help = 'Select the color for the annotation names')
parser.add_argument('--database-directories-file', metavar = 'FILE', default = "%s/.bob_bio_databases.txt" % os.environ["HOME"], help = 'An optional file, where database directories are stored (to avoid changing the database configurations)')
parser.add_argument('-o', '--output', help = "If given, it will save the plots in this output file instead of showing them. This option is useful when you don't have a display server (ssh).")
parser.add_argument('--self-test', action='store_true', help=argparse.SUPPRESS)
......@@ -73,9 +74,10 @@ def main(command_line_parameters=None):
# open figure
if not args.self_test:
from matplotlib import pyplot
pyplot.ion()
pyplot.show()
figure = pyplot.figure()
if args.output is None:
pyplot.ion()
pyplot.show()
pyplot.figure()
for f in files:
# load image
......@@ -126,7 +128,10 @@ def main(command_line_parameters=None):
pyplot.gca().set_aspect("equal")
pyplot.gca().autoscale(tight=True)
pyplot.pause(0.001)
if args.output is None:
pyplot.pause(0.001)
else:
pyplot.savefig(args.output)
input_text = "Press Enter to continue to the next image (or Ctrl-C + Enter to exit)"
if sys.version_info >= (3, 0):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment