python:基于YOLO框架和遥感图像的目标检测

作者:CSDN @ 养乐多

本文将介绍如何通过YOLO框架和遥感图像进行目标检测的代码。


文章目录


一、数据集下载与格式转换

下载数据集之后,统一转换为YOLO格式。

1.1 NWPU VHR-10(73.1 MB)

下载链接:https://opendatalab.com/OpenDataLab/NWPU_VHR-10

标注信息

python 复制代码
class_id_to_name = {
    0: "飞机",
    1: "船只",
    2: "储油罐",
    3: "棒球场",
    4: "网球场",
    5: "篮球场",
    6: "跑道场地",
    7: "港口",
    8: "桥梁",
    9: "车辆"
}

NWPU VHR-10 数据集转换为YOLO格式,参考博客《python:将 NWPU_VHR-10 遥感目标检测数据集转换成 YOLO 格式》。并配置data.yaml文件。

1.2 DIOR(7.06 GB)

下载链接:https://opendatalab.com/OpenDataLab/DIOR

DIOR数据集转换为YOLO格式,参考博客《YOLO:VOC格式数据集转换为YOLO数据集格式》,并配置data.yaml文件。

1.3 配置data.yaml

data.yaml 文件中,写入以下几行代码。保存。

修改训练图片路径和验证路径。

python 复制代码
train: E:\\DataSet\\NWPU-YOLO\\train
val: E:\\DataSet\\NWPU-YOLO\\val
nc: 10
#names: ['飞机', '船只', '储油罐', '棒球场', '网球场', '篮球场', '跑道场地', '港口', '桥梁', '车辆']
names: ['Airplane', 'Ship', 'Oil Tank', 'Baseball Field', 'Tennis Court', 'Basketball Court', 'Runway', 'Harbor', 'Bridge', 'Vehicle']

二、训练

python 复制代码
from ultralytics import YOLO
from swanlab.integration.ultralytics import add_swanlab_callback

if __name__ == '__main__':
    model = YOLO("yolov8n.pt")
    add_swanlab_callback(model, project='training_project')
    model.train(data="D:/data.yaml", epochs=10, workers=0) 
    metrics = model.val()
    # model.predict("../测试图片/00011.jpg", imgsz=640, save=True, device=0)
    path = model.export(format="onnx")

三、训练结果

训练100次。

标注数据1:

预测结果1:

标注数据2:

预测数据2:

相关推荐
这里有鱼汤3 分钟前
“对象”?对象你个头!——Python世界观彻底崩塌的一天
后端·python
尘浮72812 分钟前
60天python训练计划----day59
开发语言·python
wh393315 分钟前
使用Python将PDF转换成word、PPT
python·pdf·word
船长@Quant32 分钟前
数学视频动画引擎Python库 -- Manim Voiceover 语音服务 Speech Services
python·数学·manim·动画引擎·语音旁白
好开心啊没烦恼2 小时前
Python 数据分析:计算,分组统计1,df.groupby()。听故事学知识点怎么这么容易?
开发语言·python·数据挖掘·数据分析·pandas
lljss20203 小时前
Python11中创建虚拟环境、安装 TensorFlow
开发语言·python·tensorflow
空中湖3 小时前
tensorflow武林志第二卷第九章:玄功九转
人工智能·python·tensorflow
CodeCraft Studio4 小时前
CAD文件处理控件Aspose.CAD教程:使用 Python 将绘图转换为 Photoshop
python·photoshop·cad·aspose·aspose.cad
Python×CATIA工业智造6 小时前
Frida RPC高级应用:动态模拟执行Android so文件实战指南
开发语言·python·pycharm
onceco6 小时前
领域LLM九讲——第5讲 为什么选择OpenManus而不是QwenAgent(附LLM免费api邀请码)
人工智能·python·深度学习·语言模型·自然语言处理·自动化