added datetime to logs

Author: davidpagnon

Pose2Sim/Pose2Sim.py

@@ -46,6 +46,7 @@
 import time
 from copy import deepcopy
 import logging, logging.handlers
+from datetime import datetime
 
 
 ## AUTHORSHIP INFORMATION
@@ -193,11 +194,13 @@ def calibration(config=None):
 
     # Set up logging
     setup_logging(session_dir)  
+    currentDateAndTime = datetime.now()
 
     # Run calibration
     calib_dir = [os.path.join(session_dir, c) for c in os.listdir(session_dir) if ('Calib' or 'calib') in c][0]
     logging.info("\n\n---------------------------------------------------------------------")
     logging.info("Camera calibration")
+    logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
     logging.info("---------------------------------------------------------------------")
     logging.info(f"\nCalibration directory: {calib_dir}")
     start = time.time()
@@ -231,18 +234,20 @@ def poseEstimation(config=None):
 
     # # Set up logging
     # session_dir = os.path.realpath(os.path.join(config_dicts[0].get('project').get('project_dir'), '..', '..'))
-    # setup_logging(session_dir)    
+    # setup_logging(session_dir)
 
     # # Batch process all trials
     # for config_dict in config_dicts:
     #     start = time.time()
+    #     currentDateAndTime = datetime.now()
     #     project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
     #     seq_name = os.path.basename(project_dir)
     #     frame_range = config_dict.get('project').get('frame_range')
     #     frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
 
     #     logging.info("\n\n---------------------------------------------------------------------")
-    #     logging.info("Camera synchronization")
+    #     logging.info("Pose estimation")
+    #     logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
     #     logging.info("---------------------------------------------------------------------")
     #     logging.info(f"\nProject directory: {project_dir}")
 
@@ -280,13 +285,15 @@ def synchronization(config=None):
     # # Batch process all trials
     # for config_dict in config_dicts:
     #     start = time.time()
+    #     currentDateAndTime = datetime.now()
     #     project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
     #     seq_name = os.path.basename(project_dir)
     #     frame_range = config_dict.get('project').get('frame_range')
     #     frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
 
     #     logging.info("\n\n---------------------------------------------------------------------")
     #     logging.info("Camera synchronization")
+    #     logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
     #     logging.info("---------------------------------------------------------------------")
     #     logging.info(f"\nProject directory: {project_dir}")
 
@@ -324,13 +331,15 @@ def personAssociation(config=None):
     # Batch process all trials
     for config_dict in config_dicts:
         start = time.time()
+        currentDateAndTime = datetime.now()
         project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
         seq_name = os.path.basename(project_dir)
         frame_range = config_dict.get('project').get('frame_range')
         frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
 
         logging.info("\n\n---------------------------------------------------------------------")
         logging.info(f"Associating persons for {seq_name}, for {frames}.")
+        logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
         logging.info("---------------------------------------------------------------------")
         logging.info(f"\nProject directory: {project_dir}")
 
@@ -367,13 +376,15 @@ def triangulation(config=None):
     # Batch process all trials
     for config_dict in config_dicts:
         start = time.time()
+        currentDateAndTime = datetime.now()
         project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
         seq_name = os.path.basename(project_dir)
         frame_range = config_dict.get('project').get('frame_range')
         frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
 
         logging.info("\n\n---------------------------------------------------------------------")
         logging.info(f"Triangulation of 2D points for {seq_name}, for {frames}.")
+        logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
         logging.info("---------------------------------------------------------------------")
         logging.info(f"\nProject directory: {project_dir}")
 
@@ -413,13 +424,15 @@ def filtering(config=None):
 
     # Batch process all trials
     for config_dict in config_dicts:
+        currentDateAndTime = datetime.now()
         project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
         seq_name = os.path.basename(project_dir)
         frame_range = config_dict.get('project').get('frame_range')
         frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
 
         logging.info("\n\n---------------------------------------------------------------------")
         logging.info(f"Filtering 3D coordinates for {seq_name}, for {frames}.")
+        logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
         logging.info("---------------------------------------------------------------------")
         logging.info(f"\nProject directory: {project_dir}\n")
 
@@ -450,13 +463,15 @@ def markerAugmentation(config=None):
 
     for config_dict in config_dicts:
         start = time.time()
+        currentDateAndTime = datetime.now()
         project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
         seq_name = os.path.basename(project_dir)
         frame_range = config_dict.get('project').get('frame_range')
         frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
 
         logging.info("\n\n---------------------------------------------------------------------")
         logging.info(f"Augmentation process for {seq_name}, for {frames}.")
+        logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
         logging.info("---------------------------------------------------------------------")
         logging.info(f"\nProject directory: {project_dir}\n")
 
@@ -466,7 +481,6 @@ def markerAugmentation(config=None):
         logging.info(f'\nAugmentation took {end - start:.2f} s.')
 
 
-
 def opensimProcessing(config=None):
     '''
     Uses OpenSim to run scaling based on a static trc pose
@@ -497,6 +511,7 @@ def opensimProcessing(config=None):
 
     # # Batch process all trials
     # for config_dict in config_dicts:
+    #     currentDateAndTime = datetime.now()
     #     start = time.time()
     #     project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
     #     seq_name = os.path.basename(project_dir)
@@ -508,6 +523,7 @@ def opensimProcessing(config=None):
     #     #     logging.info(f"Scaling model with <STATIC TRC FILE>.")
     #     # else:
     #     #     logging.info(f"Running inverse kinematics <MOTION TRC FILE>.")
+    #     logging.info(f"On {currentDateAndTime.strftime('%A %d. %B %Y, %H:%M:%S')}")
     #     logging.info("---------------------------------------------------------------------")
     #     logging.info(f"\nOpenSim output directory: {project_dir}")