fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added datetime to logs

Browse Source
This commit is contained in:
davidpagnon 2024-03-05 02:59:25 +01:00
parent 3ffd39fe17
commit c04df78697
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Pose2Sim/Pose2Sim.py
View File

@ -46,6 +46,7 @@ import os
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()
@ -236,13 +239,15 @@ def poseEstimation(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("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,6 +285,7 @@ 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')
@ -287,6 +293,7 @@ def synchronization(config=None):
# 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,6 +331,7 @@ 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')
@ -331,6 +339,7 @@ def personAssociation(config=None):
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,6 +376,7 @@ 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')
@ -374,6 +384,7 @@ def triangulation(config=None):
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,6 +424,7 @@ 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')
@ -420,6 +432,7 @@ def filtering(config=None):
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,6 +463,7 @@ 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')
@ -457,6 +471,7 @@ def markerAugmentation(config=None):
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}")