aboutsummaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
authorMichael Foiani <sotech117@michaels-mbp-4.devices.brown.edu>2022-05-09 04:10:56 -0400
committerMichael Foiani <sotech117@michaels-mbp-4.devices.brown.edu>2022-05-09 04:10:56 -0400
commit0aa9638226fb90ba5320e84f46b6559cc255cedd (patch)
tree40f7f60fd70333685a8eedb737c69662a7c3ed81 /main.py
parentc2627361f1fb1f07db757d43c094c285c6ab1cee (diff)
don't show plot
Diffstat (limited to 'main.py')
-rw-r--r--main.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/main.py b/main.py
index 3649309a..26a40b31 100644
--- a/main.py
+++ b/main.py
@@ -102,10 +102,10 @@ def main():
# save image in output folder
final_image = save_tensor_as_image(result_tensor, 'out/results/', ARGS.savefile)
- # show the final result :)
+ # store + show the final result :)
plt.imshow(final_image)
- print('\nSaved the final result to out/results/{}. Showing the final result now.'.format(ARGS.savefile))
- plt.show()
+ print('\nSaved the final result to out/results/{}.'.format(ARGS.savefile))
+ # plt.show()
ARGS = parse_args()