Cross-line-detection/README.md

代码说明

导入必要的库:

python

`import cv2

from ultralytics import YOLO`

导入 OpenCV 库用于图像处理和显示。

导入 Ultralytics 的 YOLO 库用于目标检测。

定义边界线:

python

`def define_line(image):

height, width = image.shape[:2]

line_y = int(height * 0.5)

return [(0, line_y), (width, line_y)]`

这个函数用于定义视频中的一个边界线。这里定义的是视频中间位置的一条水平线。

检查是否越线:

python

`def check_crossing(bounding_box, line):

y1, _ = bounding_box[0]

y2, _ = bounding_box[1]

_, ly = line[0]

return y1 <= ly <= y2 or y2 <= ly <= y1`

这个函数用于检查目标的边界框（bounding box）是否跨越了指定的边界线。如果边界框的任意一部分在边界线上方或下方，则认为该目标越线。

加载模型:

python

1model = YOLO('yolov8n.pt')

加载预训练的 YOLOv8 模型。在这个例子中使用的是 yolov8n.pt，这是一个轻量级的模型。

读取视频:

python

video_path = 'path/to/video.mp4'

cap = cv2.VideoCapture(video_path)

打开一个视频文件，准备逐帧读取。

主循环:

python

深色版本

while cap.isOpened(): ret, frame = cap.read() if not ret: break

这个循环会一直读取视频直到视频结束。

目标检测:

python

results = model(frame)[0]

使用 YOLO 模型对每一帧进行目标检测，并获取结果。

定义边界线:

python

line = define_line(frame)

对当前帧定义边界线。

画出边界线:

python

1cv2.line(frame, line[0], line[1], (0, 255, 0), 2)

在视频帧上绘制绿色的边界线。

检测并标记越线目标:

python

`1for box in results.boxes.xyxy:

x1, y1, x2, y2 = box.astype(int)`

# 检查是否越线

`if check_crossing(((x1, y1), (x2, y2)), line):`

   ` color = (0, 0, 255) ` # 红色表示越线

else: color = (255, 0, 0) # 蓝色表示未越线

# 画出边界框

cv2.rectangle(frame, (x1, y1), (x2, y2), color, 2)

遍历检测到的目标，并根据是否越线改变边界框的颜色。

显示帧:

python

深色版本

cv2.imshow('Frame', frame)

显示带有边界框和边界线的视频帧。

退出条件:

python

if cv2.waitKey(1) & 0xFF == ord('q'): break

如果用户按下 'q' 键，则退出循环。

释放资源:

python

cap.release() cv2.destroyAllWindows()

释放视频捕获对象，并销毁所有窗口