YOLOv8-AM骨折检测项目使用教程

1. 项目目录结构及介绍

本项目是基于YOLOv8的骨折检测改进版本，项目目录结构如下：

Fracture_Detection_Improved_YOLOv8/
├── GRAZPEDWRI-DX/               # 原始数据集目录
│   ├── data/
│   │   ├── images/              # 图片目录
│   │   ├── labels/              # 标签目录
│   │   └── meta.yaml            # 数据集元信息文件
├── imgaug.py                    # 数据增强脚本
├── requirements.txt             # 项目依赖文件
├── split.py                     # 数据集分割脚本
├── start_train.py               # 训练启动脚本
├── ultralytics/                 # YOLOv8框架相关目录
│   ├── cfg/                     # 配置文件目录
│   └── models/                  # 模型文件目录
├── .gitignore                   # git忽略文件
├── LICENSE                      # 许可证文件
└── README.md                    # 项目说明文件

• GRAZPEDWRI-DX/: 存放原始数据集，包括图片和标签。
• imgaug.py: 用于对训练集进行数据增强。
• requirements.txt: 列出了项目运行所需的Python包。
• split.py: 用于将数据集分割为训练集、验证集和测试集。
• start_train.py: 用于启动训练过程的脚本。
• ultralytics/: 包含YOLOv8框架的配置文件和模型文件。
• .gitignore: 指定git应该忽略的文件和目录。
• LICENSE: 项目的开源许可证文件。
• README.md: 提供项目的基本信息和使用说明。

2. 项目的启动文件介绍

start_train.py 是启动训练过程的主要脚本。以下是脚本的一些基本用法：

python start_train.py --model ./ultralytics/cfg/models/v8/yolov8m.yaml --data_dir ./GRAZPEDWRI-DX/data/meta.yaml

• --model: 指定使用的模型配置文件路径。
• --data_dir: 指定数据集元信息文件路径。

3. 项目的配置文件介绍

项目中的配置文件主要位于 ultralytics/cfg/ 目录下。以下是几个重要的配置文件：

• default.yaml: 包含了训练和测试的默认设置，如模型路径、数据路径、训练轮数、批处理大小等。
• yolov8m.yaml, yolov8m_ECA.yaml 等：具体的模型配置文件，定义了模型的架构、参数、训练策略等。

在开始训练之前，需要确保 default.yaml 文件中的设置正确，特别是数据集的路径和模型的相关配置。

确保所有依赖都已安装，然后可以通过修改配置文件来调整训练过程，以适应不同的需求。