尧图网站设计 尧图网站设计YAOTU DESIGN
ARTICLE DETAIL

资讯详情

深耕网站设计与一线实操的经验洞察。

遥感航拍农业异常数据集 农田高质量航拍图像,其中每张图像由 RGB 和近红外 (NIR) 通道组成,分辨率高达每像素 10 厘米。

遥感航拍农业异常数据集 农田高质量航拍图像,其中每张图像由 RGB 和近红外 (NIR) 通道组成,分辨率高达每像素 10 厘米。 遥感航拍农业异常数据集收集3,432 个农田的 94,986 张高质量航拍图像其中每张图像由 RGB 和近红外 (NIR) 通道组成分辨率高达每像素 10 厘米。注释了对农民最重要的九种类型的田间异常模式重复播种、干旱、车辙印营养缺乏、播种机漏薄、风暴损坏、洪涝、水道、杂草簇异常共20GB数据集介绍数据集概述数据集名称遥感航拍农业异常数据集图像数量共94,986张高质量航拍图像图像类型每张图像由RGB和近红外NIR通道组成图像分辨率每像素10厘米异常类型共9种分别为重复播种overseeding、干旱drought、车辙印rutting、营养缺乏nutrient_deficiency、播种机漏薄planter_skip、风暴损坏storm_damage、洪涝flooding、水道waterway、杂草簇异常weed_cluster数据量约20GB数据集划分已划分好训练集、验证集和测试集数据集结构假设你的数据集已经按照以下结构组织深色版本agriculture_anomaly_dataset/├── images/│ ├── train/│ ├── val/│ └── test/└── labels/├── train/├── val/└── test/每个文件夹中包含对应的图像文件和标签文件。确保所有图像文件都是.jpg格式而标签文件是.txt格式并且它们的名字与对应的图像文件相同。数据集配置文件创建一个数据集配置文件如agriculture_anomaly_dataset.yaml该文件定义了数据集的基本信息包括路径、类别等。示例配置如下yaml深色版本训练和验证的数据集路径train: agriculture_anomaly_dataset/images/trainval: agriculture_anomaly_dataset/images/valtest: agriculture_anomaly_dataset/images/test类别名称names:0: overseeding1: drought2: rutting3: nutrient_deficiency4: planter_skip5: storm_damage6: flooding7: waterway8: weed_cluster类别数量nc: 9训练模型安装依赖确保你的开发环境中安装了必要的软件和库。YOLOv8是基于PyTorch框架的因此你需要安装Python以及PyTorch。安装Python推荐3.7或更高版本安装PyTorch你可以从PyTorch官方网站获取安装命令根据你的系统配置选择合适的安装方式。克隆YOLOv8的官方仓库到本地并安装项目所需的其他依赖bash深色版本git clone https://github.com/ultralytics/ultralytics.gitcd ultralyticspip install -r requirements.txt2. 训练模型在完成上述准备工作后你可以开始训练模型了。打开终端进入YOLOv8项目的根目录运行训练命令bash深色版本python ultralytics/yolo/v8/detect/train.py --data agriculture_anomaly_dataset.yaml --cfg yolov8.yaml --weights yolov8x.pt --batch-size 16 --epochs 100这里–data 参数指定了数据集配置文件的路径。–cfg 参数指定了模型配置文件。–weights 参数用于指定预训练权重的路径这有助于加速训练过程并提高最终模型的性能。–batch-size 和 --epochs 分别设置了批量大小和训练轮数。模型评估训练完成后可以通过验证集来评估模型的性能。YOLOv8提供了方便的命令来进行模型评估bash深色版本python ultralytics/yolo/v8/detect/val.py --data agriculture_anomaly_dataset.yaml --weights runs/train/exp/weights/best.pt这里best.pt 是训练过程中保存的最佳模型权重文件。模型推理你可以使用训练好的模型进行推理检测新的图像中的异常。示例命令如下bash深色版本python ultralytics/yolo/v8/detect/predict.py --source path/to/your/image.jpg --weights runs/train/exp/weights/best.pt --conf 0.25这里–source 参数指定了要检测的图像路径。–conf 参数设置了置信度阈值低于该阈值的检测结果将被忽略。注意事项数据增强为了提高模型的泛化能力可以考虑使用数据增强技术如随机裁剪、翻转、颜色抖动等。超参数调整根据训练过程中观察到的损失值和验证集上的性能适当调整学习率、批量大小等超参数。硬件资源如果显存不足可以减少批量大小或使用更小的模型变体。多尺度训练可以尝试多尺度训练以提高模型对不同尺度目标的检测能力。进一步优化数据预处理确保图像质量和尺寸一致可以使用图像增强技术提高模型的鲁棒性。模型选择根据实际需求选择合适的YOLOv8模型变体如yolov8s、yolov8m、yolov8l等。多GPU训练如果有多块GPU可以使用多GPU训练来加速训练过程。示例代码数据集转换如果需要假设你的标签文件已经是YOLO格式可以直接跳过这一步。如果需要从其他格式如COCO格式转换可以参考以下代码python深色版本import osimport jsondef convert_coco_to_yolo(coco_annotation_file, output_dir, image_dir):with open(coco_annotation_file, ‘r’) as f:coco_data json.load(f)image_id_to_filename {image[id]: image[file_name] for image in coco_data[images]} category_id_to_name {category[id]: category[name] for category in coco_data[categories]} category_name_to_id {v: k for k, v in category_id_to_name.items()} for annotation in coco_data[annotations]: image_id annotation[image_id] image_filename image_id_to_filename[image_id] image_path os.path.join(image_dir, image_filename) image cv2.imread(image_path) image_height, image_width, _ image.shape category_id annotation[category_id] category_name category_id_to_name[category_id] category_id category_name_to_id[category_name] bbox annotation[bbox] x_center (bbox[0] bbox[2] / 2) / image_width y_center (bbox[1] bbox[3] / 2) / image_height width bbox[2] / image_width height bbox[3] / image_height yolo_annotation f{category_id} {x_center} {y_center} {width} {height} output_file os.path.join(output_dir, os.path.splitext(image_filename)[0] .txt) with open(output_file, a) as f: f.write(yolo_annotation \n)示例用法convert_coco_to_yolo(‘path/to/coco/annotations.json’, ‘path/to/output/labels’, ‘path/to/images’)2. 训练脚本确保你已经将数据集和配置文件准备好然后运行以下训练脚本bash深色版本进入YOLOv8项目目录cd ultralytics训练模型python ultralytics/yolo/v8/detect/train.py --data agriculture_anomaly_dataset.yaml --cfg yolov8.yaml --weights yolov8x.pt --batch-size 16 --epochs 1003. 评估脚本训练完成后评估模型性能bash深色版本python ultralytics/yolo/v8/detect/val.py --data agriculture_anomaly_dataset.yaml --weights runs/train/exp/weights/best.pt4. 推理脚本使用训练好的模型进行推理bash深色版本python ultralytics/yolo/v8/detect/predict.py --source path/to/your/image.jpg --weights runs/train/exp/weights/best.pt --conf 0.25希望以上信息对你有所帮助
返回列表