镜像名称

羊羊小栈-YOLO目标检测大模型分析系统

镜像简介

镜像已预置训练代码、样例数据集及 YOLO 基础模型（yolov8n、yolo11n、yolo12n、yolo26n），并配置好名为 py312 的 Python 虚拟环境。可直接上传新数据集开始 YOLO 模型训练。

镜像内代码目录结构

/workspace/other/
└── model_train/
    └── detect/
        ├── code/
        │   ├── predict.py    # 模型预测脚本
        │   ├── train.py      # 模型训练脚本
        │   └── val.py        # 模型验证脚本
        ├── dataset/
        │   ├── small_dataset/ # 小型数据集（用于快速测试或调试）
        │   │   ├── data.yaml # 配置文件
        │   │   ├── test/     # 测试集
        │   │   ├── train/    # 训练集
        │   │   └── valid/    # 验证集
        ├── output/  # 训练、验证、预测文件的输出路径
        └── weights/ #放置yolo基础模型权重
            ├── yolo11n.pt    
            ├── yolo12n.pt    
            ├── yolo26n.pt    
            └── yolov8n.pt

代码使用方法

1.激活python环境: 
conda activate py12

2.进入训练代码根目录:
cd /workspace/other/model_train/detect/

3. 执行训练脚本: 
python code/train.py

4.执行验证脚本: 
python code/val.py

5.执行预测脚本: 
python code/predict.py