feat: 开始任务前删除上次的日志文件

pull/21/head
梁浩 4 months ago
parent 378d5fec32
commit 16f64ed3b8

@ -7,6 +7,7 @@ import os
import subprocess
import logging
import shutil
import glob
from datetime import datetime
logging.basicConfig(
@ -72,6 +73,16 @@ def run_evaluate_task(task_id, clean_ref_dir, clean_output_dir,
# 设置日志文件
log_dir = AlgorithmConfig.LOGS_DIR
os.makedirs(log_dir, exist_ok=True)
# 清除旧日志
old_logs = glob.glob(os.path.join(log_dir, f'evaluate_task_{task_id}_*.log'))
for old_log in old_logs:
try:
os.remove(old_log)
logger.info(f"Removed old log file: {old_log}")
except Exception as e:
logger.warning(f"Failed to remove old log file {old_log}: {e}")
log_file = os.path.join(
log_dir,
f'evaluate_task_{task_id}_{datetime.now().strftime("%Y%m%d_%H%M%S")}.log'

@ -102,6 +102,16 @@ def run_finetune_task(task_id, finetune_method, train_images_dir,
# 设置日志文件
log_dir = AlgorithmConfig.LOGS_DIR
os.makedirs(log_dir, exist_ok=True)
# 清除旧日志
old_logs = glob.glob(os.path.join(log_dir, f'finetune_{finetune_method}_task_{task_id}_*.log'))
for old_log in old_logs:
try:
os.remove(old_log)
logger.info(f"Removed old log file: {old_log}")
except Exception as e:
logger.warning(f"Failed to remove old log file {old_log}: {e}")
log_file = os.path.join(
log_dir,
f'finetune_{finetune_method}_task_{task_id}_{datetime.now().strftime("%Y%m%d_%H%M%S")}.log'

@ -8,6 +8,7 @@ import os
import subprocess
import logging
import shutil
import glob
from datetime import datetime
logging.basicConfig(
@ -96,6 +97,16 @@ def run_heatmap_task(task_id, original_image_path, perturbed_image_path,
# 设置日志文件
log_dir = AlgorithmConfig.LOGS_DIR
os.makedirs(log_dir, exist_ok=True)
# 清除旧日志
old_logs = glob.glob(os.path.join(log_dir, f'heatmap_task_{task_id}_*.log'))
for old_log in old_logs:
try:
os.remove(old_log)
logger.info(f"Removed old log file: {old_log}")
except Exception as e:
logger.warning(f"Failed to remove old log file {old_log}: {e}")
log_file = os.path.join(
log_dir,
f'heatmap_task_{task_id}_{datetime.now().strftime("%Y%m%d_%H%M%S")}.log'

@ -88,6 +88,16 @@ def run_perturbation_task(task_id, algorithm_code, epsilon, input_dir, output_di
# 设置日志文件
log_dir = AlgorithmConfig.LOGS_DIR
os.makedirs(log_dir, exist_ok=True)
# 清除旧日志
old_logs = glob.glob(os.path.join(log_dir, f'perturbation_{algorithm_code}_task_{task_id}_*.log'))
for old_log in old_logs:
try:
os.remove(old_log)
logger.info(f"Removed old log file: {old_log}")
except Exception as e:
logger.warning(f"Failed to remove old log file {old_log}: {e}")
log_file = os.path.join(
log_dir,
f'perturbation_{algorithm_code}_task_{task_id}_{datetime.now().strftime("%Y%m%d_%H%M%S")}.log'

Loading…
Cancel
Save