```python import os import time import logging from ultralytics import YOLO # 导入YOLO模块用于训练模型 # 定义项目根目录路径 project_root = r'C:\Users\13451\PycharmProjects\ultralytics-8.3.163' # 定义缓存文件路径 train_cache_path = os.path.join(project_root, 'train.cache') val_cache_path = os.path.join(project_root, 'val.cache') # 定义模型配置文件路径 model_config_path = os.path.join(project_root, r'ultralytics\cfg\models\11\yolo11.yaml') # 定义数据集配置文件路径 data_yaml_path = os.path.join(project_root, 'data.yaml') # 定义模型保存路径 model_save_path = os.path.join(project_root, 'yolo11n_best.pt') # 定义日志文件目录 log_dir = os.path.join(project_root, 'info') # 创建日志目录（如果不存在） os.makedirs(log_dir, exist_ok=True) # 生成日志文件名，格式为脚本名称+时间戳 log_file_name = f"train_yolo11_{time.strftime('%Y%m%d_%H%M%S')}.log" log_file_path = os.path.join(log_dir, log_file_name) # 配置日志记录器，输出到控制台和日志文件 logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(message)s', handlers=[ logging.FileHandler(log_file_path, encoding='utf-8'), # 写入日志文件 logging.StreamHandler() # 输出到控制台 ] ) # 获取日志记录器实例 logger = logging.getLogger(__name__) def delete_cache_files(): """删除缓存文件 train.cache 和 val.cache""" if os.path.exists(train_cache_path): os.remove(train_cache_path) logger.info(f"缓存文件 train.cache 已删除") else: logger.info(f"未找到缓存文件 train.cache，跳过删除") if os.path.exists(val_cache_path): os.remove(val_cache_path) logger.info(f"缓存文件 val.cache 已删除") else: logger.info(f"未找到缓存文件 val.cache，跳过删除") def train_yolo_model(): """调用YOLO模型进行训练""" try: logger.info("开始初始化YOLO模型") model = YOLO(model_config_path) # 使用自定义模型配置文件初始化YOLO模型 logger.info("开始训练模型") results = model.train( data=data_yaml_path, # 数据集配置文件路径 epochs=100, # 总训练轮数 imgsz=640, # 输入图像尺寸 workers=1, # 数据加载线程数 batch=16, # 每批次训练样本数 device=0, # 使用GPU设备0 lr0=0.001, # 初始学习率 resume=False, # 不启用断点续训 pretrained=False, # 不使用预训练模型 exist_ok=True, # 允许覆盖已有文件 name='yolo11n_custom_train', # 设置训练结果目录名称 val=True # 启用验证集 ) logger.info("训练完成，开始保存模型") model.save(model_save_path) # 保存训练完成的模型 logger.info(f"模型已保存至路径: {model_save_path}") except Exception as e: logger.error(f"训练过程中发生异常: {e}") raise if __name__ == '__main__': """主程序入口""" try: logger.info("开始执行YOLOv11训练流程") delete_cache_files() # 删除缓存文件 train_yolo_model() # 执行模型训练 logger.info("训练流程执行完成") except Exception as e: logger.error(f"主程序执行过程中发生错误: {e}") ```