一、目标检测
import cv2
from ultralytics import YOLO
# 加载 YOLOv8 模型
model = YOLO('yolov8n.pt')
video_path =0 #摄像头ip地址
cap = cv2.VideoCapture(video_path)
while cap.isOpened():
status, frame = cap.read()
if not status:
break
results = model.predict(source=frame)
result = results[0]
anno_frame = result.plot()
cv2.imshow("V8", anno_frame)
# 设置延迟以减慢视频播放速度
if cv2.waitKey(30) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
二、关键点检测
import cv2
from ultralytics import YOLO
# 加载 YOLOv8 模型
model = YOLO('yolov8n-pose.pt')
video_path =0 #摄像头ip地址
cap = cv2.VideoCapture(video_path)
while cap.isOpened():
status, frame = cap.read()
if not status:
break
results = model.predict(source=frame)
result = results[0]
anno_frame = result.plot()
cv2.imshow("V8", anno_frame)
# 设置延迟以减慢视频播放速度
if cv2.waitKey(30) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
三、分割
import cv2
from ultralytics import YOLO
# 加载 YOLOv8 模型
model = YOLO('yolov8n-seg.pt')
video_path =0 #摄像头ip地址
cap = cv2.VideoCapture(video_path)
while cap.isOpened():
status, frame = cap.read()
if not status:
break
results = model.predict(source=frame)
result = results[0]
anno_frame = result.plot()
cv2.imshow("V8", anno_frame)
# 设置延迟以减慢视频播放速度
if cv2.waitKey(30) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()