from ultralytics import YOLO
import os
import cv2 as cv
model = YOLO("yolov8s.pt")
os.environ["OPENCV_FFMPEG_CAPTURE_OPTIONS"] = "rtsp_transport;udp"
cap = cv.VideoCapture('rtsp://210.99.70.120:1935/live/cctv001.stream')
while cap.isOpened():
ret, frame = cap.read()
if not ret:
print("Can't receive frame")
break
# cv.imshow('frame', frame) 영상표시
results = model.predict(source=frame, show=True, device="0") # yolo 표시
if cv.waitKey(1) == ord('q'):
break
cap.release()
cv.destroyAllWindows()
'Python' 카테고리의 다른 글
| yolo8 & python 3.10 이용한 객체 인식 #6 사람만 찾아서 바운딩 박스 컬러 변경 (0) | 2023.05.16 |
|---|---|
| yolo8 & python 3.10 이용한 객체 인식 #5 label 숨기기 (파라메터) (0) | 2023.05.12 |
| yolo8 & python 3.10 이용한 객체 인식 #3 사람만 감지 (0) | 2023.05.12 |
| yolo8 & python 3.10 이용한 객체 인식 #3 커스텀 감지(포트홀) (0) | 2023.05.11 |
| yolo8 & python 3.10 이용한 객체 인식 #2 웹캠 사용하기 (0) | 2023.05.10 |