yolo石油泄露数据集

YOLO石油泄露数据集是一个真实场景的高质量图片数据集，用于石油泄露检测。该数据集包含1000张图片，每张图片都有对应的voc、coco和yolo三种格式标签，可以直接用于YOLO系列的目标检测。此外，该数据集还附赠YOLO环境部署、训练案例教程和数据集划分脚本，可以根据需求自行划分训练集、验证集、测试集。

以下是使用YOLOv5进行石油泄露检测的示例代码：

import torch
import cv2

# 加载YOLOv5模型
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

# 加载图片
img = cv2.imread('path/to/image.jpg')

# 进行目标检测
results = model(img)

# 打印检测结果
print(results.pandas().xyxy[0])