--- a/tomo_find_center.py	Wed Apr 13 15:53:06 2022 +0000
+++ b/tomo_find_center.py	Wed Apr 13 16:02:44 2022 +0000
@@ -4,6 +4,7 @@
 
 import sys
 import argparse
+import tracemalloc
 
 from tomo import Tomo
 
@@ -34,6 +35,9 @@
             type=argparse.FileType('w'), default=sys.stdout, help='Log file')
     args = parser.parse_args()
 
+    # Starting memory monitoring
+    tracemalloc.start()
+
     # Set basic log configuration
     logging_format = '%(asctime)s : %(levelname)s - %(module)s : %(funcName)s - %(message)s'
     log_level = 'INFO'
@@ -71,6 +75,12 @@
             'set_range' : args.set_range, 'set_step' : args.set_step}
     tomo.findCenters(galaxy_param)
 
+    # Displaying memory usage
+    logging.info(f'Memory usage: {tracemalloc.get_traced_memory()}')
+
+    # stopping memory monitoring
+    tracemalloc.stop()
+
 if __name__ == "__main__":
     __main__()