|
|
|
|
@ -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'
|
|
|
|
|
|