计算机视觉深度学习框架YOLOV8模型如何训练水下生物检测数据集 通过训练的权重推理识别检测海底生物检测数据集中的海参海胆扇贝海星鱼类潜水员海龟等

计算机视觉深度学习框架YOLOV8模型如何训练水下生物检测数据集 通过训练的权重推理识别检测海底生物检测数据集中的海参海胆扇贝海星鱼类潜水员海龟等

计算机视觉深度学习框架YOLOV8模型如何训练水下生物检测数据集

水下目标检测数据集10类别 YOLO格式 14,000张高清图片

水下目标检测数据集信息表

类别标签对照表（共10类）

数据集划分统计

✅增强数据说明：另附未参与训练的数据增强样本（如旋转、模糊、亮度变化等），可用于模型鲁棒性测试或迁移学习。

为了使用YOLOv8训练和推理一个包含 10 类海洋目标的水下高清图像数据集（共 14,000 张图片），我们将从环境搭建、数据准备、模型训练、推理到评估提供完整的技术流程与代码。

✅ 一、环境配置

1. 安装 CUDA 驱动（GPU 支持）

确保你的系统已安装 NVIDIA 显卡驱动和 CUDA Toolkit：

nvidia-smi

输出应显示 GPU 信息。推荐 CUDA 版本：11.8 或 12.1。

2. 安装 Anaconda

下载并安装 Anaconda（Python 3.9+ 推荐）。

3. 创建 Python 虚拟环境

conda create-nunderwater_yolopython=3.9conda activate underwater_yolo

—

4. 安装依赖项

# 安装 PyTorch（以 CUDA 11.8 为例）pipinstalltorch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118# 安装 YOLOv8 官方库pipinstallultralytics opencv-python numpy matplotlib tqdm scikit-learn# 可选：TensorBoard 可视化pipinstalltensorboard

验证安装：

python-c"import torch; print(torch.cuda.is_available())"# 应输出 True

✅ 二、数据集结构与data.yaml配置

数据目录结构

underwater_dataset/ ├── images/ │ ├── train/ # 9800 张 │ └── val/ # 4200 张 ├── labels/ │ ├── train/ # 对应的 .txt 标注文件 │ └── val/ # 对应的 .txt 标注文件 └── data.yaml # 数据集配置文件

创建data.yaml

在项目根目录创建data.yaml文件：

# data.yamltrain:./underwater_dataset/images/trainval:./underwater_dataset/images/val# 类别数量nc:10# 类别名称（英文）names:-holothurian-echinus-scallop-starfish-fish-corals-diver-cuttlefish-turtle-jellyfish# 中文映射（仅用于可视化）names_zh:-海参-海胆-扇贝-海星-鱼类-珊瑚-潜水员-乌贼-海龟-水母

✅ 三、调用官方预训练模型并开始训练

使用 YOLOv8 官方预训练模型（如yolov8x.pt）

fromultralyticsimportYOLO# 加载预训练模型（可选：yolov8n, yolov8s, yolov8m, yolov8l, yolov8x）model=YOLO('yolov8x.pt')# 下载自动进行# 开始训练results=model.train(data='data.yaml',# 数据配置文件epochs=150,# 训练轮数batch=16,# 批次大小（根据显存调整）imgsz=1280,# 输入图像尺寸（适合小目标检测）optimizer='AdamW',# 优化器lr0=0.001,# 初始学习率augment=True,# 启用内置数据增强hsv_h=0.015,# 色彩扰动hsv_s=0.7,hsv_v=0.4,degrees=10.0,# 随机旋转translate=0.1,scale=0.5,shear=0.0,flipud=0.0,fliplr=0.5,mosaic=1.0,mixup=0.1,copy_paste=0.1,project='runs/train',# 保存路径name='underwater_yolov8x',# 实验名称save=True,save_period=10,# 每10轮保存一次cache=False# 若内存充足可设为 True 加速加载)

💡提示：

• 若显存不足，降低batch或使用yolov8m.pt
• imgsz=1280有助于提升小目标检测精度（水下目标常较小）

✅ 四、推理代码（单图/批量/视频）

1. 单张图像推理

fromultralyticsimportYOLOfromPILimportImage# 加载训练好的模型model=YOLO('runs/train/underwater_yolov8x/weights/best.pt')# 推理results=model('test_image.jpg')# 显示结果forrinresults:im_array=r.plot()# 绘制边界框和标签im=Image.fromarray(im_array[...,::-1])# BGR to RGBim.show()

2. 批量图像推理（文件夹）

importosfrompathlibimportPath source_dir='path/to/test_images/'output_dir='runs/detect/predictions'os.makedirs(output_dir,exist_ok=True)results=model(source_dir,save=True,project=output_dir,name='batch_predict')print(f"批量推理完成，结果保存在{output_dir}")

3. 视频检测

results=model.predict(source='underwater_video.mp4',save=True,project='runs/detect/video',name='result',show=False)print("视频检测完成")

4. 摄像头实时检测（OpenCV）

cap=cv2.VideoCapture(0)whilecap.isOpened():ret,frame=cap.read()ifnotret:breakresults=model(frame)annotated_frame=results[0].plot()cv2.imshow('YOLOv8 Real-time Detection',annotated_frame)ifcv2.waitKey(1)&0xFF==ord('q'):breakcap.release()cv2.destroyAllWindows()

✅ 五、模型评估（验证集）

# 在验证集上评估metrics=model.val(data='data.yaml',split='val',batch=16,imgsz=1280,save_json=False,project='runs/val',name='underwater_eval')# 输出关键指标print(f"mAP@0.5:{metrics.box.map50:.4f}")print(f"mAP@0.5:0.95:{metrics.box.map:.4f}")print(f"Precision:{metrics.box.p:.4f}")print(f"Recall:{metrics.box.r:.4f}")# 每类 APfori,nameinenumerate(model.names):print(f"{name}: AP@0.5 ={metrics.box.ap[i]:.4f}")

✅ 六、可视化与结果分析

1. 绘制中文标签（可选）

# 修改模型类名显示为中文model.names=['海参','海胆','扇贝','海星','鱼类','珊瑚','潜水员','乌贼','海龟','水母']results=model('test.jpg')results[0].plot()# 此时显示中文标签

2. 使用 TensorBoard 查看训练曲线

tensorboard--logdirruns/train/underwater_yolov8x

✅ 七、导出模型（ONNX / TensorRT / CoreML）可根据电脑系统

# 导出为 ONNX（通用部署格式）model.export(format='onnx',dynamic=True,opset=13)# 导出为 TensorRT（NVIDIA GPU 加速）model.export(format='engine',half=True,dynamic=True)# 导出为 TFLite（移动端）model.export(format='tflite')

以上文字及代码仅供参考学习使用。